[debug]
Browse files
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 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
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 |
|