harshm121 commited on
Commit
a805467
1 Parent(s): 517eea6
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -23,10 +23,10 @@ mtmodelpath = './checkpoints/sid_1-500_mtteacher.pth'
23
  mtmodel.load_state_dict(torch.load(mtmodelpath, map_location=torch.device('cpu')))
24
  mtmodel.eval()
25
 
26
- # m3lmodel = LinearFusionMaskedConsistencyMixBatch("mit_b2", args, num_classes=13, pretrained=False)
27
- # m3lmodelpath = './checkpoints/sid_1-500_m3lteacher.pth'
28
- # m3lmodel.load_state_dict(torch.load(m3lmodelpath, map_location=torch.device('cpu')))
29
- # m3lmodel.eval()
30
 
31
 
32
 
 
23
  mtmodel.load_state_dict(torch.load(mtmodelpath, map_location=torch.device('cpu')))
24
  mtmodel.eval()
25
 
26
+ m3lmodel = LinearFusionMaskedConsistencyMixBatch("mit_b2", args, num_classes=13, pretrained=False)
27
+ m3lmodelpath = './checkpoints/sid_1-500_m3lteacher.pth'
28
+ m3lmodel.load_state_dict(torch.load(m3lmodelpath, map_location=torch.device('cpu')))
29
+ m3lmodel.eval()
30
 
31
 
32