FBCNN icon indicating copy to clipboard operation
FBCNN copied to clipboard

Trained models usage.

Open mobil0ft opened this issue 1 year ago • 0 comments

Greetings, Sorry for bothering, but how can I run the trained model? I tried to create another .py script for trained color.pth model (similar to Color/Real scripts) But getting an error when tryin to run it.

File "main_test_fbcnn_latest_g.py", line 105, in main() File "main_test_fbcnn_latest_g.py", line 55, in main model.load_state_dict(torch.load(model_path), strict=True) File "D:\Users\Vasily\anaconda3\envs\FBCNN\lib\site-packages\torch\serialization.py", line 593, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "D:\Users\Vasily\anaconda3\envs\FBCNN\lib\site-packages\torch\serialization.py", line 763, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: could not find MARK

How to solve this issue, could you, please, help? Thank you in advance~

mobil0ft avatar Feb 28 '23 21:02 mobil0ft