Update train.py

train.py

@@ -24,21 +24,17 @@ print("Torchvision Version: ",torchvision.__version__)
 # Much of the code is a modified version of the code available at
 # https://pytorch.org/tutorials/beginner/finetuning_torchvision_models_tutorial.html
 
-
-# nohup python train.py > logs/empty_cell_aug_28032024.txt 2>&1 &
-# echo $! > logs/save_pid.txt
-
 parser = argparse.ArgumentParser('arguments for training')
 
-parser.add_argument('--tr_empty_folder', type=str, default="/data/taulukot/solukuvat/empty/train/",
+parser.add_argument('--tr_empty_folder', type=str, default="/path/to/empty/train/images",
                     help='path to training data with empty images')
-parser.add_argument('--val_empty_folder', type=str, default="/data/taulukot/solukuvat/empty/val/",
+parser.add_argument('--val_empty_folder', type=str, default="/path/to/empty/validation/images",
                     help='path to validation data with empty images')
-parser.add_argument('--tr_ok_folder', type=str, default="/data/taulukot/solukuvat/ok/train/",
+parser.add_argument('--tr_ok_folder', type=str, default="/path/to/non-empty/train/images",
                     help='path to training data with ok images')
-parser.add_argument('--val_ok_folder', type=str, default="/data/taulukot/solukuvat/ok/val/",
+parser.add_argument('--val_ok_folder', type=str, default="/path/to/non-empty/validation/images",
                     help='path to validation data with ok images')
-parser.add_argument('--results_folder', type=str, default="results/28032024_aug/",
+parser.add_argument('--results_folder', type=str, default="./results/",
                     help='Folder for saving training results.')
 parser.add_argument('--save_model_path', type=str, default="./models/",
                     help='Path for saving model file.')
@@ -56,11 +52,11 @@ parser.add_argument('--random_seed', type=int, default=8765,
                     help='Number used for initializing random number generation.')
 parser.add_argument('--early_stop_threshold', type=int, default=3,
                     help='Threshold value of epochs after which training stops if validation accuracy does not improve.')
-parser.add_argument('--save_model_format', type=str, default='torch',
+parser.add_argument('--save_model_format', type=str, default='onnx',
                     help='Defines the format for saving the model.')
 parser.add_argument('--augment_choice', type=str, default=None,
                     help='Defines which image augmentation(s) are used. Defaults to randomly selected augmentations.')
-parser.add_argument('--model_name', type=str, default='aug_b32_lr0001',
+parser.add_argument('--model_name', type=str, default='test_model',
                     help='Current date.')
 parser.add_argument('--date', type=str, default=time.strftime("%d%m%Y"),
                     help='Current date.')