summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-04-04 12:44:40 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-04-04 12:44:40 +0800
commitcb05de36f5ffd8584d78c6776dbe90e21abff25a (patch)
tree4eaf5708fd11596470ffdff7d1045b73546fdd3f /config.py
parentf6f133fa7b926ce0c7d28bbf0ba4de41b3708d4a (diff)
Remove triplet sampler
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