From 0a1bc7a154c55a5ae7cec116d322b27d7df33d0d Mon Sep 17 00:00:00 2001 From: tsugumi-sys Date: Wed, 3 Jan 2024 10:55:53 +0900 Subject: [PATCH] Fix broken evaluator (SA-ConvLSTM) --- .../moving_mnist_pipeline/self_attention_memory_convlstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipelines/moving_mnist_pipeline/self_attention_memory_convlstm.py b/pipelines/moving_mnist_pipeline/self_attention_memory_convlstm.py index 6abbbd6..201cbf4 100644 --- a/pipelines/moving_mnist_pipeline/self_attention_memory_convlstm.py +++ b/pipelines/moving_mnist_pipeline/self_attention_memory_convlstm.py @@ -88,7 +88,7 @@ def main(): ### print("Evaluating ...") evaluator = Evaluator( - model=None, + model=model, test_dataloader=data_loaders.test_dataloader, save_dir_path="./tmp/evaluate", )