欢迎光临散文网 会员登陆 & 注册

90天学会GAN--week2--如何复现别人的模型?

2023-06-11 13:08 作者:弱弱的小汤汤  | 我要投稿

方式非常简单,用的是 ablation study的思路:

也就是先实现一个最基本的步骤,然后一步一步往中间加内容。

写在前面 

由于是第一次复现模型,在这里写一个模板方便之后构建模型: 

首先是 option.py: 

在这个文件中,我们会输入一些在模型中会用的参数:

之后是model.py:

在之后是real_model.py ,也就是模型的部分,这里按照模型不同可能有差异,这里就先略过这个部分。 

然后是dataloader:

最后是training 和 validation 的部分:

至此,模型就写完啦。

很抱歉拖更了这么久,但是复现别人的模型确实是一个很难的点。

算是第一个瓶颈吧。(づ ̄3 ̄)づ╭❤~

90天学会GAN--week2--如何复现别人的模型?的评论 (共 条)

分享到微博请遵守国家法律