anton/mlpractical
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add runner for batch normalized version

Browse Source
This commit is contained in:
Anton Lydike 2024-11-19 09:47:18 +00:00
parent cb5c6f4e19
commit 94d3a1d484
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_mlp_framework/arg_extractor.py
View File

@ -118,7 +118,7 @@ def get_args():
"--block_type", "--block_type",
type=str, type=str,
default="conv_block", 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)", "(This argument will be useful in running experiments to debug your network)",
) )
args = parser.parse_args() args = parser.parse_args()

1
run_vgg_38_bn.sh Normal file
View File

@ -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