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