--- a
+++ b/code/3_train.py
@@ -0,0 +1,29 @@
+import config
+from utils_model import train_resnet
+
+# Training the ResNet.
+print("\n\n+++++ Running 3_train.py +++++")
+train_resnet(batch_size=config.args.batch_size,
+             checkpoints_folder=config.args.checkpoints_folder,
+             classes=config.classes,
+             color_jitter_brightness=config.args.color_jitter_brightness,
+             color_jitter_contrast=config.args.color_jitter_contrast,
+             color_jitter_hue=config.args.color_jitter_hue,
+             color_jitter_saturation=config.args.color_jitter_saturation,
+             device=config.device,
+             learning_rate=config.args.learning_rate,
+             learning_rate_decay=config.args.learning_rate_decay,
+             log_csv=config.log_csv,
+             num_classes=config.num_classes,
+             num_layers=config.args.num_layers,
+             num_workers=config.args.num_workers,
+             path_mean=config.path_mean,
+             path_std=config.path_std,
+             pretrain=config.args.pretrain,
+             resume_checkpoint=config.args.resume_checkpoint,
+             resume_checkpoint_path=config.resume_checkpoint_path,
+             save_interval=config.args.save_interval,
+             num_epochs=config.args.num_epochs,
+             train_folder=config.args.train_folder,
+             weight_decay=config.args.weight_decay)
+print("+++++ Finished running 3_train.py +++++\n\n")