From 683fba76862980d33e0982c9844403e6c8a01fcd Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Tue, 14 Dec 2021 13:51:00 -0800
Subject: [PATCH] Add drop args to benchmark.py

---
 benchmark.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/benchmark.py b/benchmark.py
index ccd9b4fa..f1604a04 100755
--- a/benchmark.py
+++ b/benchmark.py
@@ -199,7 +199,11 @@ class BenchmarkRunner:
             num_classes=kwargs.pop('num_classes', None),
             in_chans=3,
             global_pool=kwargs.pop('gp', 'fast'),
-            scriptable=torchscript)
+            scriptable=torchscript,
+            drop_rate=kwargs.pop('drop', 0.),
+            drop_path_rate=kwargs.pop('drop_path', None),
+            drop_block_rate=kwargs.pop('drop_block', None),
+        )
         self.model.to(
             device=self.device,
             dtype=self.model_dtype,