1.使用了Visdom来展示一些训练中的图片以及损失等等
训练速度很慢,在cmd中使用nvidia-smi命令可以看到的GPU的功率和利用率非常低
注释掉每次保存以后,功率有了明显提升以及利用率上来了,但是还是半天才会输出一个epoch信息。
这之后虽然功率和利用率上来了,但是一个epoch信息还是输出的很慢,阅读了一些博客,对数据的多线程读取,以及固定内存
我将原来的num_workers设置为了8,pin_memory设置为了True,只能说杯水车薪,并没有显著提高,还是有1/3的功率没有用起来。训练打印epoch的速度可能快了一点。