Skip to content

Commit 34f382f

Browse files
committed
move dataconfig before script, scripting killing metadata now (PyTorch 1.12? just nvfuser?)
1 parent beef62e commit 34f382f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchmark.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,11 +225,12 @@ def __init__(
225225
self.num_classes = self.model.num_classes
226226
self.param_count = count_params(self.model)
227227
_logger.info('Model %s created, param count: %d' % (model_name, self.param_count))
228+
229+
data_config = resolve_data_config(kwargs, model=self.model, use_test_size=not use_train_size)
228230
self.scripted = False
229231
if torchscript:
230232
self.model = torch.jit.script(self.model)
231233
self.scripted = True
232-
data_config = resolve_data_config(kwargs, model=self.model, use_test_size=not use_train_size)
233234
self.input_size = data_config['input_size']
234235
self.batch_size = kwargs.pop('batch_size', 256)
235236

0 commit comments

Comments
 (0)