summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-04-04 12:45:36 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-04-04 12:45:36 +0800
commit6a8824e4fb8bdd1f3e763b78b765830788415cfb (patch)
tree1853fce051cb9c46c7d253d7ddcec3cde0478899 /config.py
parent2f4516549c7b9ebfaf650c6a71e4da43cd3372fa (diff)
parentcb05de36f5ffd8584d78c6776dbe90e21abff25a (diff)
Merge branch 'disentangling_only' into disentangling_only_py3.8
# Conflicts: # models/model.py # utils/configuration.py # utils/sampler.py
Diffstat (limited to 'config.py')
-rw-r--r--config.py8
1 files changed, 3 insertions, 5 deletions
diff --git a/config.py b/config.py
index dc4e0ba..cf76618 100644
--- a/config.py
+++ b/config.py
@@ -11,7 +11,7 @@ config: Configuration = {
# Recorde disentangled image or not
'image_log_on': True,
# The number of subjects for validating (Part of testing set)
- 'val_size': 10,
+ 'val_size': 20,
},
# Dataset settings
'dataset': {
@@ -36,10 +36,8 @@ config: Configuration = {
},
# Dataloader settings
'dataloader': {
- # Batch size (pr, k)
- # `pr` denotes number of persons
- # `k` denotes number of sequences per person
- 'batch_size': (4, 6),
+ # Batch size
+ 'batch_size': 16,
# Number of workers of Dataloader
'num_workers': 4,
# Faster data transfer from RAM to GPU if enabled