Spaces:
Sleeping
Sleeping
from easydict import EasyDict | |
cuda = False | |
multi_gpu = False | |
main_config = dict( | |
exp_name='pendulum_ibc_seed0', | |
env=dict( | |
evaluator_env_num=5, | |
act_scale=True, | |
n_evaluator_episode=5, | |
stop_value=-250, | |
), | |
policy=dict( | |
cuda=cuda, | |
model=dict(obs_shape=3, action_shape=1, stochastic_optim=dict( | |
type='mcmc', | |
cuda=cuda, | |
)), | |
learn=dict( | |
multi_gpu=multi_gpu, | |
train_epoch=15, | |
batch_size=256, | |
optim=dict(learning_rate=1e-5, ), | |
learner=dict(hook=dict(log_show_after_iter=1000)), | |
), | |
collect=dict( | |
data_type='hdf5', | |
data_path='./pendulum_sac_data_generation/expert_demos.hdf5', | |
collector_logit=False, | |
), | |
eval=dict(evaluator=dict(eval_freq=-1, )), | |
), | |
) | |
pendulum_ibc_config = EasyDict(main_config) | |
main_config = pendulum_ibc_config | |
pendulum_ibc_create_config = dict( | |
env=dict( | |
type='pendulum', | |
import_names=['dizoo.classic_control.pendulum.envs.pendulum_env'], | |
), | |
env_manager=dict(type='base'), | |
policy=dict( | |
type='ibc', | |
import_names=['ding.policy.ibc'], | |
), | |
) | |
pendulum_ibc_create_config = EasyDict(pendulum_ibc_create_config) | |
create_config = pendulum_ibc_create_config | |