try to use my own dataset on colab but fail.

[zbeibei31]

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!