From 5a063855dbecb8f1a86ad25d9e61a9c8b63312b3 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 25 Mar 2021 12:23:23 +0800
Subject: Bug fixes and refactoring

1. Correct trained model signature
2. Move `val_size` to system config
---
 utils/configuration.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'utils')

diff --git a/utils/configuration.py b/utils/configuration.py
index 157d249..5a5bc0c 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -8,13 +8,13 @@ class SystemConfiguration(TypedDict):
     CUDA_VISIBLE_DEVICES: str
     save_dir: str
     image_log_on: bool
+    val_size: int
 
 
 class DatasetConfiguration(TypedDict):
     name: str
     root_dir: str
     train_size: int
-    val_size: int
     num_sampled_frames: int
     truncate_threshold: int
     discard_threshold: int
-- 
cgit v1.2.3