diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index c31546da1d70a4318fedefd697b558c9fc2700f0..2240ad971d59b17d58b1a461680acf0d8607c8ad 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -778,12 +778,12 @@ class ESPCN: self.build_evaluation() self.restore(ckpt_dir) - def run_evaluate(self, filename, ckpt_dir): - self.setup_eval_pipeline(filename) + def run_evaluate(self, nda_lr, param, ckpt_dir): + # self.setup_eval_pipeline(filename) self.build_model() self.build_training() self.build_evaluation() - self.do_evaluate(ckpt_dir) + self.do_evaluate(nda_lr, param, ckpt_dir) if __name__ == "__main__":