lambdanet / backup /deblur /src /train-raft.sh
hyliu's picture
Upload folder using huggingface_hub
8ec10cf verified
#!/bin/bash
bash ./prepare.sh
python launch.py --n_GPUs 4 main.py --batch_size 12 --save_dir RAFTNET_L1_R6_test --n_scales 6 \
--loss 1*L1 --model RaftNet --n_resblocks 10 --amp true --decay_gamma 0.5