Пример #1
0
_g_conf.TYPE = "task"
_g_conf.LABSMOOTH = 0
_g_conf.L1_WEIGHT = 0.1
_g_conf.PRETRAINED = "RECON"
_g_conf.PRETRAINED = "IL"
_g_conf.TRAIN_TYPE = 'LSGAN'
_g_conf.SEED = 5
_g_conf.CLAMP = 0.01
_g_conf.N_CRITIC = 5
_g_conf.TASK_ADV_WEIGHT = 0.1
_g_conf.IF_AUG = True
_g_conf.IF_POOL = True
_g_conf.IF_AUG_SRC = True
_g_conf.IF_AUG_TGT = True
_g_conf.TARGET_DOMAIN_PATH = './'
_g_conf.DECODER_RECON_PATH = './'
_g_conf.IL_AGENT_PATH = './'
_g_conf.FINETUNE = True
_g_conf.MODEL_PATH = './'
_g_conf.MIN_LEARNING_RATE = 0.00002
_g_conf.ADJUST_LR = 1

_g_conf.MODEL_NAME = 'coil_icra'

_g_conf.TRAINING_SCHEDEULE = [[50000, 0.5], [100000, 0.5 * 0.5],
                              [150000, 0.5 * 0.5 * 0.5],
                              [200000, 0.5 * 0.5 * 0.5 * 0.5],
                              [250000, 0.5 * 0.5 * 0.5 * 0.5 * 0.5]
                              ]  # Number of iterations, multiplying factor
"""#### Simulation Related Parameters ####"""
Пример #2
0
_g_conf.LABSMOOTH = 0
_g_conf.L1_WEIGHT = 0.1
_g_conf.PRETRAINED = "RECON"
_g_conf.PRETRAINED = "IL"
_g_conf.TRAIN_TYPE = 'LSGAN'
_g_conf.SEED = 5
_g_conf.CLAMP = 0.01
_g_conf.N_CRITIC = 5
_g_conf.TASK_ADV_WEIGHT = 0.1
_g_conf.IF_AUG_SRC = True
_g_conf.IF_AUG_TGT = True
_g_conf.IF_AUG = False
_g_conf.IF_POOL = True

_g_conf.IL_AGENT_PATH = 'Encoder_IL.pth'
_g_conf.DECODER_RECON_PATH = 'Decoder_Recon_Aug.pth'
_g_conf.TARGET_DOMAIN_PATH = '/datatmp/Datasets/UNIT_LW1toEW12/trainB'
"""#### Network Related Parameters ####"""

_g_conf.MODEL_NAME = 'coil_icra'
_g_conf.TRAINING_SCHEDEULE = [[50000, 0.5], [100000, 0.5 * 0.5],
                              [150000, 0.5 * 0.5 * 0.5],
                              [200000, 0.5 * 0.5 * 0.5 * 0.5],
                              [250000, 0.5 * 0.5 * 0.5 * 0.5 * 0.5]
                              ]  # Number of iterations, multiplying factor
#TODO check how to use this part

_g_conf.LEARNING_RATE = 0.0002  # First
_g_conf.BRANCH_LOSS_WEIGHT = [0.95, 0.95, 0.95, 0.95, 0.05]
_g_conf.VARIABLE_WEIGHT = {
    'Steer': 0.5,