#!/bin/bash


srun python finetune.py \
--model taming \
--modelpath checkpoints/2021-04-03T19-39-50_cin_transformer/ \
--dataset codes-imagenet \
--datapath taming/taming_gen_images_original_decoder_50k/codes \
--dataset_size 50000 \
--loss hard-to-soft-with-ae-no-reg \
--nb_epochs 50 \
--lr 0.0005 \
--n_step_lr 2 \
--augs none \
--augs_schedule 5,5,20,20 \
--batch_size_per_gpu 8 \
--idempotence_loss_weight 1.0 \
--outdir taming/finetune/taming_gen_10k_bs8_lr5e-4_step2_reg_no_aug552020/ \
