# 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 | |