From 34f382f8f6583a80cb0a169c275bf0806d95ca06 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 1 Jul 2022 14:50:36 -0700 Subject: [PATCH] move dataconfig before script, scripting killing metadata now (PyTorch 1.12? just nvfuser?) --- benchmark.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmark.py b/benchmark.py index f348fcb9..1362eeab 100755 --- a/benchmark.py +++ b/benchmark.py @@ -225,11 +225,12 @@ class BenchmarkRunner: self.num_classes = self.model.num_classes self.param_count = count_params(self.model) _logger.info('Model %s created, param count: %d' % (model_name, self.param_count)) + + data_config = resolve_data_config(kwargs, model=self.model, use_test_size=not use_train_size) self.scripted = False if torchscript: self.model = torch.jit.script(self.model) self.scripted = True - data_config = resolve_data_config(kwargs, model=self.model, use_test_size=not use_train_size) self.input_size = data_config['input_size'] self.batch_size = kwargs.pop('batch_size', 256)