From 8703b11deee847d423f2d93fb49810df8b14abe1 Mon Sep 17 00:00:00 2001 From: Johannes Fischer Date: Thu, 28 Oct 2021 13:31:30 +0200 Subject: [PATCH] import evaluation --- scratch/johannes/gail_options_image.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scratch/johannes/gail_options_image.py b/scratch/johannes/gail_options_image.py index ac1676a..03849d5 100644 --- a/scratch/johannes/gail_options_image.py +++ b/scratch/johannes/gail_options_image.py @@ -24,6 +24,7 @@ from tqdm import tqdm from src.policies.options import OptionsCnnPolicy from src.gail.options import OptionsEnv, LLOptions, HLOptions, RenderOptions from src.gail.train import train_discriminator, train_generator +from src.evaluation.evaluation import Evaluation model_name = 'gail_options_image' Env = NRasterizedRandomAgent