未验证 提交 162effb5 编写于 作者: H haoyuying 提交者: GitHub

Remove the evaluation in the image coloring task

上级 8fdb482d
......@@ -21,4 +21,4 @@ if __name__ == '__main__':
model.set_config(classification=False, prob=0.125)
optimizer = paddle.optimizer.Adam(learning_rate=0.00001, parameters=model.parameters())
trainer = Trainer(model, optimizer, checkpoint_dir='img_colorization_ckpt_reg_1')
trainer.train(color_set, epochs=101, batch_size=25, eval_dataset=color_set, log_interval=10, save_interval=10)
trainer.train(color_set, epochs=101, batch_size=25, log_interval=10, save_interval=10)
......@@ -11,4 +11,4 @@ if __name__ == "__main__":
styledata = MiniCOCO(transform)
optimizer = paddle.optimizer.Adam(learning_rate=0.0001, parameters=model.parameters())
trainer = Trainer(model, optimizer, checkpoint_dir='test_style_ckpt')
trainer.train(styledata, epochs=101, batch_size=4, eval_dataset=styledata, log_interval=10, save_interval=10)
\ No newline at end of file
trainer.train(styledata, epochs=101, batch_size=4, log_interval=10, save_interval=10)
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册