AI417UPM
/

A4_4313204_Fardows

Model card Files Files and versions

Fardows11 commited on Feb 18

Commit

78e8e09

·

verified ·

1 Parent(s): a143d73

Update hubconf.py

Files changed (1) hide show

hubconf.py +2 -2

hubconf.py CHANGED Viewed

@@ -14,7 +14,7 @@ def DummyNet(pretrained=True):
             map_location="cpu"
         )
         model.load_state_dict(state_dict)
-        model.eval()
     return model
@@ -26,5 +26,5 @@ def VanillaNet(pretrained=True):
             map_location="cpu"
         )
         model.load_state_dict(state_dict)
-        model.eval()
     return model

             map_location="cpu"
         )
         model.load_state_dict(state_dict)
     return model
             map_location="cpu"
         )
         model.load_state_dict(state_dict)
     return model