Closed JonathanLeban closed 2 years ago
Linked to #45.
Thank you for your work.
I had tested your code changes. But my cuda 11.6 gou environment could not solve the problem without adding following patch.
add caling model.to(device) after calling model.load_state_dict()
Thank you for your work.
I had tested your code changes. But my cuda 11.6 gou environment could not solve the problem without adding following patch.
add caling model.to(device) after calling model.load_state_dict()
Thank you for the feedback @adakoda. Could you please pull down #52 and rerun to see if this resolves your issue? I was able to reproduce the issue as stated before, and this new PR resolves it for me.
Proposed change(s)
Describe the changes made in this PR.
Fixed a cpu/gpu error made during the inference.