不知道读了多少遍了加油
-
把数据集解压一下
cd EDSR_train
然后`` -
更改Then, change the dir_data argument in src/option.py to the place where DIV2K images are located.
改成 -
我们建议您在训练前对图像进行预处理。这一步将解码所有png文件并将它们保存为二进制文件。在第一次运行时使用——ext sep_reset参数。您可以跳过解码部分,使用已保存的二进制文件,并使用—ext sep参数。看不懂什么意思。
Saving...
Traceback (most recent call last):
File "main.py", line 33, in <module>
main()
File "main.py", line 28, in main
t.test()
File "/home/zhj/EDSR-PyTorch/src/trainer.py", line 123, in test
self.ckp.save(self, epoch, is_best=(best[1][0, 0] + 1 == epoch))
File "/home/zhj/EDSR-PyTorch/src/utility.py", line 88, in save
trainer.loss.plot_loss(self.dir, epoch)
File "/home/zhj/EDSR-PyTorch/src/loss/__init__.py", line 111, in plot_loss
plt.plot(axis, self.log[:, i].numpy(), label=label)
File "/home/zhj/anaconda3/envs/EDSR/lib/python3.6/site-packages/matplotlib/pyplot.py", line 2763, in plot
is not None else {}), **kwargs)
File "/home/zhj/anaconda3/envs/EDSR/lib/python3.6/site-packages/matplotlib/axes/_axes.py", line 1646, in plot
lines = [*self._get_lines(*args, data=data, **kwargs)]
File "/home/zhj/anaconda3/envs/EDSR/lib/python3.6/site-packages/matplotlib/axes/_base.py", line 216, in __call__
yield from self._plot_args(this, kwargs)
File "/home/zhj/anaconda3/envs/EDSR/lib/python3.6/site-packages/matplotlib/axes/_base.py", line 342, in _plot_args
raise ValueError(f"x and y must have same first dimension, but "
ValueError: x and y must have same first dimension, but have shapes (0,) and (1,)
更换成1.0.0版本的代码就好了。
5.开始跑起来了 乘以2版本的