def create_config():
"""
Set up experimental parameters here.
"""
config = DotMap()
config.peg_data = "training_dataset_brijen/peg_transfer"
config.random_data = "training_dataset_brijen/random"
config.training_data = config.random_data # which dataset to train on
config.actual_inputs = False # whether to use actual as input
config.history = 10
config.batch_size = 100
config.training_iterations = 5000
config.lr = 1e-3
config.device = "cpu" # switch to cuda if desired
config.save_dir = "log" # where to save logs and models
config.save_freq = 10 # how often to save model
config.log_freq = 10 # how often to print progress
config.validation_prob = 0.1
config.rnn = False
config.consistency_weight = 1. # set to 0 to not use cyclic consistency loss
return config
