summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-28 22:13:43 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-28 22:13:43 +0800
commitc96a6c88fa63d62ec62807abf957c9a8df307b43 (patch)
treed20b3ade49d7b16a253a23766988890b5ce59a68 /config.py
parentbbd89fda7f6bf30c9ce4a3b576c0087858b407b3 (diff)
Modify default parameters
1. Change ReLU to Leaky ReLU in decoder 2. Add 8-scale-pyramid in HPM
Diffstat (limited to 'config.py')
-rw-r--r--config.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/config.py b/config.py
index f76cea5..9072982 100644
--- a/config.py
+++ b/config.py
@@ -53,7 +53,7 @@ config: Configuration = {
# Use 1x1 convolution in dimensionality reduction
'hpm_use_1x1conv': False,
# HPM pyramid scales, of which sum is number of parts
- 'hpm_scales': (1, 2, 4),
+ 'hpm_scales': (1, 2, 4, 8),
# Global pooling method
'hpm_use_avg_pool': True,
'hpm_use_max_pool': True,