changed config file for experiment

configs/experiment1.yaml

@@ -12,5 +12,5 @@ shuffle: False
 
 # training related
 batch_size: 16
-epochs: 10
+epochs: 15
 # epochs: 02

kaggle/kernel-metadata.json

@@ -5,7 +5,7 @@
   "language": "python",
   "kernel_type": "notebook",
   "is_private": "true",
-  "enable_gpu": "false",
+  "enable_gpu": "true",
   "enable_tpu": "false",
   "enable_internet": "true",
   "dataset_sources": [],

src/scripts/train.py

@@ -48,7 +48,7 @@ def train(args):
     
     # log model to comet
     if "LOCAL_SYSTEM" not in os.environ:
-        experiment.log_model(f"{config.task}_{config.dataset}_{config.model}",model_save_path)
+        experiment.log_model(f"{config.task}_{config.dataset}_{config.model}",model_dir)
 
     # evaluate model
     print_title("\nEvaluating Model")