--- a
+++ b/code/4_test.py
@@ -0,0 +1,40 @@
+import config
+from utils_model import get_predictions
+
+# Run the ResNet on the generated patches.
+print("\n\n+++++ Running 4_test.py +++++")
+print("\n----- Finding validation patch predictions -----")
+# Validation patches.
+get_predictions(patches_eval_folder=config.args.patches_eval_val,
+                output_folder=config.args.preds_val,
+                auto_select=config.args.auto_select,
+                batch_size=config.args.batch_size,
+                checkpoints_folder=config.args.checkpoints_folder,
+                classes=config.classes,
+                device=config.device,
+                eval_model=config.eval_model,
+                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)
+print("----- Finished finding validation patch predictions -----\n")
+print("----- Finding test patch predictions -----")
+# Test patches.
+get_predictions(patches_eval_folder=config.args.patches_eval_test,
+                output_folder=config.args.preds_test,
+                auto_select=config.args.auto_select,
+                batch_size=config.args.batch_size,
+                checkpoints_folder=config.args.checkpoints_folder,
+                classes=config.classes,
+                device=config.device,
+                eval_model=config.eval_model,
+                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)
+print("----- Finished finding test patch predictions -----\n")
+print("+++++ Finished running 4_test.py +++++\n\n")