test_resnet_imagenet_mean_stddev.sh

#!/bin/bash

# add tf_models to PYTHONPATH
# Note: due to conflicts between tf_models and tf_models/gpu we will overwrite the PYTHONPATH
# and not append to it!

export PYTHONPATH=$PWD/tf_models/models
export PYTHONPATH=$PYTHONPATH:$PWD

sed -i 's/print_mean_stddev = False/print_mean_stddev = True/g' fs_coding.py
# 32 train steps with batch size 32 each -> 1024 images
python3 tf_models/models/official/r1/resnet/imagenet_main.py \
	--eval_only \
	--data_dir=datasets/imagenet \
	--resnet_size=50 \
	--model_dir=checkpoints/imagenet/resnet50/checkpoint \
	--max_train_steps=32 \
	--use_fs \
	2> resnet_imagenet_mean_stddev.txt

python extract_mean_stddev.py \
	--file_name=resnet_imagenet_mean_stddev.txt \


sed -i 's/print_mean_stddev = True/print_mean_stddev = False/g' fs_coding.py