songquanpeng/pytorch-template

训练模式

GodeWithWind opened this issue · 2 comments

GodeWithWind commented

parser.add_argument('--mode', type=str, default='train', choices=['train', 'eval', 'sample'])
请问这里的sample模式是什么呢

songquanpeng commented

mode 参数用于设置运行模式，sample 是取样模式，对于生成性质的模型有用，例如 GAN 的训练。

  • 👍1
GodeWithWind commented

好的，感谢