如何解决如何在pytorch-lightning中将模型输出作为另一个模型的输入提供?
我已经训练了一些 pytorch-lightning 模型,使用
...
checkpoint_callback = ModelCheckpoint(dirpath=lightning_checkpoints_path)
trainer = Trainer(deterministic=config.is_deterministic,max_epochs=config.num_epochs,default_root_dir=lightning_checkpoints_path,logger=logger,callbacks=[checkpoint_callback],gpus=1
)
trainer.fit(model,datamodule=datamodule)
我现在想训练另一个模型,该模型通过相同的 datamodule
,但这次输入到新模型的每个数据点都应该是已经训练过的模型的输出,在 eval
模式下运行.
我在文档中找不到任何类似的东西,在普通的 Pytorch 中做同样的事情非常麻烦,这就是我首先使用闪电的原因。
我想这样做是为了创建一个由 VAE 和分类器组成的半监督学习框架。
这是解释here。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。