alias python="/usr/bin/python3.7.5"
source scripts/env_npu.sh
nohup python3 -u main_npu_1p.py \
	"/home/data/imagenet/" \
        --lr=1.0 \
        --print-freq=10 \
        --momentum=0.9 \
        --epochs=1 \
        --workers=$(nproc) \
        --seed=49 \
        --amp \
        --loss-scale=128.0 \
        --opt-level='O2' \
        --device='npu' \
        --world-size=1 \
        --batch-size=512 > ./log/nasnet_mobile_npu_1p_bs256.log 2>&1 &