From 59ccfd7718babe94fac549fcfbfa22bb311f0bd8 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 21 Jan 2021 21:50:28 +0800 Subject: Bug fixes 1. Turn off autograd while decoding canonical and pose features 2. Change default batch size to (4, 8) --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'config.py') diff --git a/config.py b/config.py index c93fe19..991a3a6 100644 --- a/config.py +++ b/config.py @@ -33,7 +33,7 @@ config: Configuration = { # Batch size (pr, k) # `pr` denotes number of persons # `k` denotes number of sequences per person - 'batch_size': (2, 2), + 'batch_size': (4, 8), # Number of workers of Dataloader 'num_workers': 4, # Faster data transfer from RAM to GPU if enabled -- cgit v1.2.3