change to 300 classes

app/core/config.py

@@ -4,8 +4,8 @@ BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../"))
 
 I3D_WEIGHTS = os.path.join(
     BASE_DIR,
-   # "assets/I3D_300Weights/FINAL_nslt_300_iters2997_top156.14_top579.94_top1086.98.pt"
-   "assets/I3D_300Weights/FINAL_nslt_100_iters=896_top1=65.89_top5=84.11_top10=89.92 (1).pt"
+   "assets/I3D_300Weights/FINAL_nslt_300_iters2997_top156.14_top579.94_top1086.98.pt"
+   #"assets/I3D_300Weights/FINAL_nslt_100_iters=896_top1=65.89_top5=84.11_top10=89.92 (1).pt"
 )
 
 CLASS_LIST = os.path.join(

app/models/i3d_model.py

@@ -6,7 +6,7 @@ class I3DModel:
         self.device = device
 
         self.model = InceptionI3d(400, in_channels=3)
-        self.model.replace_logits(100)
+        self.model.replace_logits(300)
         self.model.load_state_dict(torch.load(weights_path, map_location=device))
         self.model.to(device)
         self.model.eval()