Skip to content

Commit

Permalink
Update retrain_mfv.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
weichenrs committed Mar 16, 2023
1 parent f273c18 commit 3f17fb8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mfv/retrain_mfv.sh
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
#gid
#CUDA_VISIBLE_DEVICES=0 python retrain_mfv.py --dataset gid --checkname MMM_DD_DATASET_MODEL_SCALE --batch-size 2 --workers 2 --epochs 20
CUDA_VISIBLE_DEVICES=0 python retrain_mfv.py --dataset gid --checkname MMM_DD_DATASET_MODEL_SCALE --batch-size 2 --workers 2 --epochs 20

#wfv
#CUDA_VISIBLE_DEVICES=0 python retrain_mfv.py --dataset wfv --checkname MMM_DD_DATASET_MODEL_SCALE --batch-size 2 --workers 2 --epochs 20

#potsdam
#CUDA_VISIBLE_DEVICES=0 python retrain_mfv.py --dataset potsdam --checkname MMM_DD_DATASET_MODEL_SCALE --batch-size 2 --workers 2 --epochs 50
#CUDA_VISIBLE_DEVICES=0 python retrain_mfv.py --dataset potsdam --checkname MMM_DD_DATASET_MODEL_SCALE --batch-size 2 --workers 2 --epochs 100
#CUDA_VISIBLE_DEVICES=0 python retrain_mfv.py --dataset potsdam --checkname MMM_DD_DATASET_MODEL_SCALE --batch-size 2 --workers 2 --epochs 100

0 comments on commit 3f17fb8

Please sign in to comment.