train.sh

# shell
# ------------------------------------------
python3 main.py \
	--dataDir='/home/lzw/Disk/data/semantic_Segmentation/human_matting/' \
	--saveDir='./result' \
	--trainData='human_matting' \
	--trainList='train.txt' \
	--load='attunet' \
	--nThreads=4 \
	--patch_size=256 \
	--train_batch=64 \
	--lr=1e-3 \
	--lrdecayType='keep' \
	--nEpochs=500 \
	--save_epoch=1 \

python3 main.py \
	--dataDir='/home/lzw/Disk/data/semantic_Segmentation/human_matting/' \
	--saveDir='./result' \
	--trainData='human_matting' \
	--trainList='train.txt' \
	--load='attunet' \
	--nThreads=4 \
	--patch_size=256 \
	--train_batch=64 \
	--lr=1e-4 \
	--lrdecayType='keep' \
	--nEpochs=1000 \
	--save_epoch=1 \
	--finetuning \

python3 main.py \
	--dataDir='/home/lzw/Disk/data/semantic_Segmentation/human_matting/' \
	--saveDir='./result' \
	--trainData='human_matting' \
	--trainList='train.txt' \
	--load='attunet' \
	--nThreads=3 \
	--patch_size=256 \
	--train_batch=64 \
	--lr=1e-5 \
	--lrdecayType='keep' \
	--nEpochs=1500 \
	--save_epoch=1 \
	--finetuning \
	--train_refine