Hugging Face's logo

Spaces:
tunne
/
rice_leaf_diseases
Sleeping

App Files Community
unknown commited on
Commit
89521e9
·
1 Parent(s): 148f991

hehe

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -42,7 +42,7 @@ import copy
42
 
43
  # Load GoogleNet model
44
  # os.makedirs('./model', exist_ok=True, mode=0o777)
45
- os.environ['TORCH_HOME'] = './models'
46
  model = models.googlenet(pretrained=True)
47
  model.fc= nn.Linear(1024, 4)
48
  model.load_state_dict(torch.load('./model_transfer_batch_2_epoch50.pt', map_location=torch.device('cpu')))
 
42
 
43
  # Load GoogleNet model
44
  # os.makedirs('./model', exist_ok=True, mode=0o777)
45
+ # os.environ['TORCH_HOME'] = './models'
46
  model = models.googlenet(pretrained=True)
47
  model.fc= nn.Linear(1024, 4)
48
  model.load_state_dict(torch.load('./model_transfer_batch_2_epoch50.pt', map_location=torch.device('cpu')))