Skip to content

Commit 06fd5c9

Browse files
committed
Set default value for use_bf16 and fixed random seed setting error
Signed-off-by: Cheng, Penghui <[email protected]>
1 parent 79db56f commit 06fd5c9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

neural_compressor/config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def __init__(self,
285285
max_trials=100,
286286
performance_only=False,
287287
reduce_range=None,
288-
extra_precisions=[],
288+
extra_precisions=["bf16"],
289289
accuracy_criterion=accuracy_criterion):
290290
self._inputs = inputs
291291
self._outputs = outputs
@@ -837,7 +837,7 @@ def __init__(
837837

838838

839839
def set_random_seed(seed: int):
840-
options.random_seed
840+
options.random_seed = seed
841841

842842

843843
def set_workspace(workspace: str):

0 commit comments

Comments
 (0)