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
---
 config.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'config.py')

diff --git a/config.py b/config.py
index 3d98263..66eab98 100644
--- a/config.py
+++ b/config.py
@@ -9,7 +9,9 @@ config: Configuration = {
         # Directory used in training or testing for temporary storage
         'save_dir': 'runs',
         # Recorde disentangled image or not
-        'image_log_on': False
+        'image_log_on': False,
+        # The number of subjects for validating (Part of testing set)
+        'val_size': 10,
     },
     # Dataset settings
     'dataset': {
@@ -19,8 +21,6 @@ config: Configuration = {
         'root_dir': 'data/CASIA-B-MRCNN-V2/SEG',
         # The number of subjects for training
         'train_size': 74,
-        # The number of subjects for validating (Part of testing set)
-        'val_size': 10,
         # Number of sampled frames per sequence (Training only)
         'num_sampled_frames': 30,
         # Truncate clips longer than `truncate_threshold`
-- 
cgit v1.2.3