2021年4月16日
训练模型的剩余时间
摘要： 1 import time 2 import datetime 3 prev_time = time.time() #epoch循环之外 4 # epoch训练循环之内 5 batches_done = (epoch - start_iter + 1) * len(data_loader) + _ 阅读全文
posted @ 2021-04-16 20:10 临近边缘 阅读(162) 评论(0) 推荐(0)
保存训练过程中的修复图
摘要： 1 damaged_images = ground_truths * masks + (1 - masks) 2 outputs_comps = ground_truths * masks + outputs * (1 - masks) 3 if count % 1000 == 0: 4 sizes 阅读全文
posted @ 2021-04-16 20:08 临近边缘 阅读(44) 评论(0) 推荐(0)
可视化loss和metrics
摘要： tensorboard 可视化 1 from tensorboardX import SummaryWriter 2 writer = SummaryWriter(args.log_dir) 3 G_loss, G_loss_list = generator_loss(input_images, m 阅读全文
posted @ 2021-04-16 20:02 临近边缘 阅读(226) 评论(0) 推荐(0)
Pytorch 断点继训
摘要： pytorch 模型的断点继训 pytorch 模型断点继训 1 checkpoint = { "model_state_dict": net.state_dict(), 2 "optimizer_state_dict": optimizer.state_dict(), 3 "epoch": epo 阅读全文
posted @ 2021-04-16 09:04 临近边缘 阅读(99) 评论(0) 推荐(0)