Fix Pytorch models
In this MR, some bugs related to PyTorch models are fixed and improved:
- add
to(device)
, - add
model.eval()
, - add
torch.no_grad()
Edited by Hatef OTROSHI
In this MR, some bugs related to PyTorch models are fixed and improved:
to(device)
,model.eval()
,torch.no_grad()