# This is a ADE20K training script in a 512x512 configuration.
# Used to test how long and how expensive this model would be to train from scratch.
import sys
sys.path.append('../lib/SPADE-master/')
from options.train_options import TrainOptions
from models.pix2pix_model import Pix2PixModel
from collections import OrderedDict
import data
from util.iter_counter import IterationCounter
from util.visualizer import Visualizer
from trainers.pix2pix_trainer import Pix2PixTrainer
opt = TrainOptions()
opt.D_steps_per_G = 1
opt.aspect_ratio = 1.0
opt.batchSize = 1
opt.beta1 = 0.0
opt.beta2 = 0.9
opt.cache_filelist_read = False
opt.cache_filelist_write = False
opt.checkpoints_dir = '/spell/checkpoints/'
opt.contain_dontcare_label = True
opt.continue_train = False
opt.crop_size = 512
opt.dataroot = '/spell/ade20k_c/ADEChallengeData2016/' # data mount point
opt.dataset_mode = "ade20k"
opt.debug = False
opt.display_freq = 100
opt.display_winsize = 512
