submission(ailorg#2)

configs/config.yaml

@@ -1,5 +1,5 @@
 batch_size: 256
-epochs: 5
+epochs: 20
 lr: 0.001
 
 device: cuda:0

main.py

@@ -10,15 +10,12 @@
 from tqdm import tqdm
 
 from src.datasets import ThingsMEGDataset
-from src.models_finetuning import BasicConvClassifier
+from src.ResNet50 import BasicConvClassifier
 from src.utils import set_seed
 
 #GPUが認識されるか確認
 print("GPU is available:", torch.cuda.is_available())
 
-#前処理の関数を定義
-# def ICA(X: torch.Tensor) -> torch.Tensor:
-
 
 @hydra.main(version_base=None, config_path="configs", config_name="config")
 def run(args: DictConfig):
@@ -52,7 +49,6 @@ def run(args: DictConfig):
         train_set.num_classes, train_set.seq_len, train_set.num_channels
     ).to(args.device)
 
-    # model = 
 
     # ------------------
     #     Optimizer
@@ -123,6 +119,5 @@ def run(args: DictConfig):
     np.save(os.path.join(logdir, "submission"), preds)
     cprint(f"Submission {preds.shape} saved at {logdir}", "cyan")
 
-#コマンドから実行した場合に実行
 if __name__ == "__main__":
     run()