train first stage python unet_train.py --data_dir /data/common/ILSVRC/Data/CLS-LOC/train --log_dir output/cls0.1_tanh --w_cls0 0.1