From 39fd755baaeccd6fcea548ef5f3542ade2ef2433 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Mon, 8 Feb 2021 08:09:41 -0700
Subject: [PATCH] New benchmark numbers

---
 codes/models/vqvae/vqvae_3_hardswitch.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/codes/models/vqvae/vqvae_3_hardswitch.py b/codes/models/vqvae/vqvae_3_hardswitch.py
index cfdef76d..ff6a950e 100644
--- a/codes/models/vqvae/vqvae_3_hardswitch.py
+++ b/codes/models/vqvae/vqvae_3_hardswitch.py
@@ -231,8 +231,15 @@ def register_vqvae3_hard_switch(opt_net, opt):
 
 
 def performance_test():
+    # For breadth=32:
+    # Custom_cuda_naive: 28.9s
+    # Torch_native: 29.2s
+    #
+    # For breadth=8
+    # Custom_cuda_naive: 18.4s
+    # Torch_native: 10s
     cfg = {
-        'mode': 'lambda2',
+        'mode': 'lambda',
         'breadth': 8,
         'hard_enabled': True,
         'dropout': 0.4