rosie

Results 1 issues of rosie

I got Runtime Error when loading pre-trained model. (./checkpoint/injection_final-002.pth) ``` File "inference.py", line 240, in handler = MegaFS(args.swap_type, args.img_root, args.mask_root) File "inference.py", line 80, in `__init__` ckpts = torch.load(ckpt_e, map_location=torch.device("cuda"...