def setup_mixins_tnb(policy, action_space, obs_space, config):
setup_mixins(policy, action_space, obs_space, config)
NoveltyValueNetworkMixin.__init__(policy, obs_space, action_space, config)
assert isinstance(config["checkpoint_dict"], str)
checkpoint_dict = json.loads(config["checkpoint_dict"])
AgentPoolMixin.__init__(policy, checkpoint_dict,
config['novelty_threshold'],
config['distance_mode'])
def setup_mixins_modified(policy, obs_space, action_space, config):
setup_mixins(policy, obs_space, action_space, config)
AddLossMixin.__init__(policy, config)
NoveltyParamMixin.__init__(policy, config)
def setup_mixins_dice(policy, obs_space, action_space, config):
setup_mixins(policy, obs_space, action_space, config)
DiversityValueNetworkMixin.__init__(policy, obs_space, action_space,
config)
discrete = isinstance(action_space, gym.spaces.Discrete)
ComputeDiversityMixin.__init__(policy, discrete)
def setup_mixins_modified(policy, obs_space, action_space, config):
AddLossMixin.__init__(policy, config)
setup_mixins(policy, obs_space, action_space, config)
def setup_mixins_dece(policy, action_space, obs_space, config):
setup_mixins(policy, action_space, obs_space, config)
NoveltyValueNetworkMixin.__init__(policy, obs_space, action_space, config)
ComputeNoveltyMixin.__init__(policy)
ConstrainNoveltyMixin.__init__(policy, config)
def before_loss_init(policy, obs_space, action_space, config):
setup_mixins(policy, obs_space, action_space, config)
FIMEmbeddingMixin.__init__(policy)