From 94d3a1d4840caf329a83ac1ad975038f2b4f5029 Mon Sep 17 00:00:00 2001 From: Anton Lydike Date: Tue, 19 Nov 2024 09:47:18 +0000 Subject: [PATCH] add runner for batch normalized version --- pytorch_mlp_framework/arg_extractor.py | 2 +- run_vgg_38_bn.sh | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) create mode 100644 run_vgg_38_bn.sh diff --git a/pytorch_mlp_framework/arg_extractor.py b/pytorch_mlp_framework/arg_extractor.py index 73810a3..2351100 100644 --- a/pytorch_mlp_framework/arg_extractor.py +++ b/pytorch_mlp_framework/arg_extractor.py @@ -118,7 +118,7 @@ def get_args(): "--block_type", type=str, default="conv_block", - help="Type of convolutional blocks to use in our network " + help="Type of convolutional blocks to use in our network" "(This argument will be useful in running experiments to debug your network)", ) args = parser.parse_args() diff --git a/run_vgg_38_bn.sh b/run_vgg_38_bn.sh new file mode 100644 index 0000000..a206f6e --- /dev/null +++ b/run_vgg_38_bn.sh @@ -0,0 +1 @@ +python pytorch_mlp_framework/train_evaluate_image_classification_system.py --batch_size 100 --seed 0 --num_filters 32 --num_stages 3 --num_blocks_per_stage 5 --experiment_name VGG_38_experiment --use_gpu True --num_classes 100 --block_type 'conv_bn' --continue_from_epoch -1