swim_new / train_swim.sh
qninhdt
u can come back later
d2c8fab
raw
history blame
588 Bytes
# export WANDB_DIR="./wandb"
# export WANDB_CACHE_DIR="./wandb"
# export WANDB_CONFIG_DIR="./wandb"
# export TMPDIR=/home/ninhnq3/temp
python swim/train.py \
data.root_dir=/cm/shared/ninhnq3/datasets/swim_data \
model=swim_gan \
logger=wandb \
logger.wandb.save_dir=/cm/shared/ninhnq3/workdirs/swim \
data.batch_size=1 \
trainer.devices=1 \
model.learning_rate=1e-5 \
trainer.max_epochs=1000 \
+trainer.val_check_interval=0.05 \
+trainer.limit_val_batches=0.005 \
callbacks.model_checkpoint.dirpath=/cm/shared/ninhnq3/checkpoints/swim_new