richzhang / colorization-pytorch

PyTorch reimplementation of Interactive Deep Colorization
https://richzhang.github.io/ideepcolor/
MIT License
598 stars 114 forks source link

Errors when running CPU only #20

Open UsmanKaisser opened 3 years ago

UsmanKaisser commented 3 years ago

Im trying to run using CPU only in a VM. I keep getting the following error during the first part of the batch script:

`[Network G] Total number of parameters : 34.187 M

No handlers could be found for logger "visdom" create web directory ./checkpoints/siggraph_class_small/web... Traceback (most recent call last): File "train.py", line 61, in model.optimize_parameters() File "/home/testing/Desktop/colorization-pytorch-master/models/pix2pix_model.py", line 193, in optimize_parameters self.forward() File "/home/testing/Desktop/colorization-pytorch-master/models/pix2pix_model.py", line 123, in forward self.fake_B_dec_max = self.netG.module.upsample4(util.decode_max_ab(self.fake_B_class, self.opt)) File "/home/testing/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 576, in getattr type(self).name, name)) AttributeError: 'SIGGRAPHGenerator' object has no attribute 'module' mkdir: cannot create directory ‘./checkpoints/siggraph_class’: File exists cp: cannot stat './checkpoints/siggraph_class_small/latest_net_G.pth': No such file or directory `