From 9001f7e13d8985b220bd218d8de716bc586dbdcf Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Fri, 26 Feb 2021 20:17:03 +0800
Subject: Update default config
---
config.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/config.py b/config.py
index 88ad371..03f2f0d 100644
--- a/config.py
+++ b/config.py
@@ -37,7 +37,7 @@ config: Configuration = {
# Batch size (pr, k)
# `pr` denotes number of persons
# `k` denotes number of sequences per person
- 'batch_size': (4, 8),
+ 'batch_size': (4, 6),
# Number of workers of Dataloader
'num_workers': 4,
# Faster data transfer from RAM to GPU if enabled
@@ -64,7 +64,7 @@ config: Configuration = {
# Embedding dimension for each part
'embedding_dims': 256,
# Triplet loss margins for HPM and PartNet
- 'triplet_margins': (0.2, 0.2),
+ 'triplet_margins': (1.5, 1.5),
},
'optimizer': {
# Global parameters
@@ -83,15 +83,15 @@ config: Configuration = {
# 'amsgrad': False,
# Local parameters (override global ones)
- 'auto_encoder': {
- 'weight_decay': 0.001
- },
+ # 'auto_encoder': {
+ # 'weight_decay': 0.001
+ # },
},
'scheduler': {
# Period of learning rate decay
'step_size': 500,
# Multiplicative factor of decay
- 'gamma': 0.9,
+ 'gamma': 1,
}
},
# Model metadata
--
cgit v1.2.3