CycleGAN icon indicating copy to clipboard operation
CycleGAN copied to clipboard

try to use my own dataset on colab but fail.

Open zbeibei31 opened this issue 3 years ago • 0 comments

I am using my own dataset on Colab. I can train my dataset but when I need to test it.

dataset [SingleDataset] was created initialize network with normal model [TestModel] was created loading the model from ./checkpoints/women2men_pretrained/latest_net_G.pth Traceback (most recent call last): File "test.py", line 47, in model.setup(opt) # regular setup: load and print networks; create schedulers File "/content/pytorch-CycleGAN-and-pix2pix/models/base_model.py", line 88, in setup self.load_networks(load_suffix) File "/content/pytorch-CycleGAN-and-pix2pix/models/base_model.py", line 192, in load_networks state_dict = torch.load(load_path, map_location=str(self.device)) File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 595, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 764, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) EOFError: Ran out of input

please help thanks!

zbeibei31 avatar Nov 23 '20 22:11 zbeibei31