-
Notifications
You must be signed in to change notification settings - Fork 5
/
train-craft-f2full-gma.sh
6 lines (6 loc) · 1.86 KB
/
train-craft-f2full-gma.sh
1
2
3
4
5
6
#!/usr/bin/env bash
env CUDA_VISIBLE_DEVICES=2,3 python3 train.py --name craft-chairs --stage chairs --validation chairs --output results/chairs/craft-f2full-gma --num_steps 120000 --lr 0.00025 --image_size 368 496 --wdecay 0.0001 --gpus 0 1 --batch_size 8 --val_freq 10000 --print_freq 100 --mixed_precision --craft --f2 full # --setrans
env CUDA_VISIBLE_DEVICES=2,3 python3 train.py --name craft-things --stage things --validation sintel --output results/things/craft-f2full-gma --restore_ckpt results/chairs/craft-f2full-gma/craft-chairs.pth --num_steps 120000 --lr 0.000125 --image_size 400 720 --wdecay 0.0001 --gpus 0 1 --batch_size 6 --val_freq 10000 --print_freq 100 --mixed_precision --craft --f2 full # --setrans
env CUDA_VISIBLE_DEVICES=2,3 python3 train.py --name craft-sintel --stage sintel --validation sintel --output results/sintel/craft-f2full-gma --restore_ckpt results/things/craft-f2full-gma/craft-things.pth --num_steps 120000 --lr 0.000125 --image_size 368 768 --wdecay 0.00001 --gamma 0.85 --gpus 0 1 --batch_size 6 --val_freq 10000 --print_freq 100 --mixed_precision --craft --f2 full # --setrans
env CUDA_VISIBLE_DEVICES=2,3 python3 train.py --name craft-kitti --stage kitti --validation kitti --output results/kitti/craft-f2full-gma --restore_ckpt results/sintel/craft-f2full-gma/craft-sintel.pth --num_steps 50000 --lr 0.000125 --image_size 288 960 --wdecay 0.00001 --gamma 0.85 --gpus 0 1 --batch_size 6 --val_freq 10000 --print_freq 100 --mixed_precision --craft --f2 full # --setrans
env CUDA_VISIBLE_DEVICES=2,3 python3 train.py --name craft-viper --stage viper --validation viper --output results/viper/craft-f2full-gma --restore_ckpt results/sintel/craft-f2full-gma/craft-sintel.pth --num_steps 50000 --lr 0.000125 --image_size 288 960 --wdecay 0.00001 --gamma 0.85 --gpus 0 1 --batch_size 6 --val_freq 10000 --print_freq 100 --mixed_precision --craft --f2 full # --setrans