Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Trained models usage. #16

Open
mobil0ft opened this issue Feb 28, 2023 · 0 comments
Open

Trained models usage. #16

mobil0ft opened this issue Feb 28, 2023 · 0 comments

Comments

@mobil0ft
Copy link

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~

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant