Skip to content

Commit 5557c73

Browse files
authored
fix logger print format (#1125)
1 parent 70a85d4 commit 5557c73

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

neural_compressor/adaptor/tensorflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,7 @@ def quantize(self, tune_cfg, model, data_loader, q_func=None):
549549
batch_size = get_model_input_shape(model)
550550
logger.warning(
551551
"Fail to forward with batch size={}, set to {} now.".
552-
format(batch_size, batch_size))
552+
format(data_loader.batch_size, batch_size))
553553
data_loader.batch(batch_size)
554554
self.quantize_config['calib_iteration'] = calib_sampling_size
555555
converted_model = GraphConverter(model,

0 commit comments

Comments
 (0)