rohithk-03 commited on
Commit
9a973b8
·
1 Parent(s): 08323cf

update gdrive url

Browse files
Files changed (1) hide show
  1. model.py +0 -2
model.py CHANGED
@@ -150,8 +150,6 @@ def check_file(image_path):
150
  "/home/user/app/checkpoint32.pth", weights_only=False, map_location=torch.device('cpu'))
151
  checkpoint = remove_module_from_checkpoint(checkpoint)
152
  model.load_state_dict(checkpoint['model_state_dict'])
153
- # optimizer.load_state_dict(checkpoint['optimizer_state_dict'])
154
- # scheduler.load_state_dict(checkpoint['scheduler_state_dict'])
155
  model.eval()
156
  model.to(device)
157
  model = nn.DataParallel(model)
 
150
  "/home/user/app/checkpoint32.pth", weights_only=False, map_location=torch.device('cpu'))
151
  checkpoint = remove_module_from_checkpoint(checkpoint)
152
  model.load_state_dict(checkpoint['model_state_dict'])
 
 
153
  model.eval()
154
  model.to(device)
155
  model = nn.DataParallel(model)