harshm121 commited on
Commit
517eea6
1 Parent(s): 0fb9d5b
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