def test_regression_maml(self):
maml_model = pose_env_maml_models.PoseEnvRegressionModelMAML(
base_model=pose_env_models.PoseEnvRegressionModel())
train_eval.train_eval_model(
t2r_model=maml_model,
input_generator_train=self._meta_record_input_generator_train,
input_generator_eval=self._meta_record_input_generator_eval,
create_exporters_fn=None)
def test_regression_maml_policy_interface(self):
t2r_model = pose_env_maml_models.PoseEnvRegressionModelMAML(
base_model=pose_env_models.PoseEnvRegressionModel(),
preprocessor_cls=preprocessors.FixedLenMetaExamplePreprocessor)
predictor = checkpoint_predictor.CheckpointPredictor(t2r_model=t2r_model)
predictor.init_randomly()
policy = meta_policies.MAMLRegressionPolicy(t2r_model, predictor=predictor)
self._test_policy_interface(policy, restore=False)