We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent beef62e commit 34f382fCopy full SHA for 34f382f
benchmark.py
@@ -225,11 +225,12 @@ def __init__(
225
self.num_classes = self.model.num_classes
226
self.param_count = count_params(self.model)
227
_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)
230
self.scripted = False
231
if torchscript:
232
self.model = torch.jit.script(self.model)
233
self.scripted = True
- data_config = resolve_data_config(kwargs, model=self.model, use_test_size=not use_train_size)
234
self.input_size = data_config['input_size']
235
self.batch_size = kwargs.pop('batch_size', 256)
236
0 commit comments