Spaces:
Runtime error
Runtime error
rohithk-03
commited on
Commit
·
9a973b8
1
Parent(s):
08323cf
update gdrive url
Browse files
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)
|