config = {}

config['input_width'] = 1024
config['input_height'] = 256

config['g_input_nc'] = 3*2
config['g_output_nc'] = 3

config['g_nfg'] = 32
config['g_layers'] = 8

config['d_input_nc'] = 12

config['d_nfg'] = 64
config['d_layers'] = 6

config['output_dir'] = './model_output'

config['adam_lr'] = 0.001
config['adam_b1'] = 0.9
config['adam_b2'] = 0.999

config['evaluation_snapshots_cnt'] = 7
config['evaluation_recursive_samples'] = 334
config['full_simulaiton_samples'] = 80

config['lambda_L1'] = 100

config['lstm'] = True
config['data_dir'] = './data/generated_data/'
config['model_name'] = 'unetLSTM'
config['use_pressure'] = False
config['model_type'] = 's'
config['cuda'] = False
config['threads'] = 4
config['batch_size'] = 3
config['seed'] = 123
config['niter'] = 100
config['epochs'] = 1
config['niter_decay'] = 100
config['lr_policy'] = 'step'
config['evaluate'] = False
config['no_train'] = False
config['model_path'] = None
config['output_dir'] = './model_output/unetLSTM32param'
config['mask'] = True
config['noise'] = True
config['crops'] = True
config['parallel'] = False
config['lstm_layers'] = 1
config['random'] = True
config['cuda_device'] = 'cuda:0'
config['batchreset'] = True
config['shuffle_batch'] = False
config['use_params'] = False
config['reset_iter'] = 1
config['random_until'] = -1
config['freezing'] = False
