Skip to content

Commit fe62213

Browse files
authored
block dataset logger info (#908)
Signed-off-by: n1ck-guo <heng.guo@intel.com>
1 parent 0c2619c commit fe62213

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

auto_round/calib_dataset.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,12 @@
1313
# limitations under the License.
1414

1515
import json
16+
import logging
1617
import random
1718
import sys
1819

20+
logging.getLogger("datasets").setLevel(logging.WARNING)
21+
1922
import torch
2023
from datasets import Dataset, Features, IterableDataset, Sequence, Value, concatenate_datasets, load_dataset
2124
from torch.utils.data import DataLoader

0 commit comments

Comments
 (0)