Running in docker FileNotFoundError: [Errno 2] No such file or directory: 'checkpoints/cm.lib'

When I run the app in Docker as described, I got: docker run --rm -it -v $PWD:/app:rw ababy/python-deepnude /bin/bash root@11dcb1effe70:/app# ls README.md gan.py input.png main.py opencv_transform readmeimgs run.py root@11dcb1effe70:/app# python main.py libpng warning: iCCP: known incorrect sRGB profile Executing phase: dress_to_correct Executing phase: correct_to_mask Traceback (most recent call last): File "main.py", line 30, in <module> main() File "main.py", line 21, in main watermark = process(dress) File "/app/run.py", line 115, in process model.initialize(opt) File "/app/gan.py", line 69, in initialize self.__load_network(self.netG) File "/app/gan.py", line 89, in __load_network network.load_state_dict(torch.load(save_path)) File "/usr/local/lib/python3.7/site-packages/torch/serialization.py", line 382, in load f = open(f, 'rb') FileNotFoundError: [Errno 2] No such file or directory: 'checkpoints/cm.lib' root@11dcb1effe70:/app#

How can I fix this?

Thanks