[distil] fix once for all general logger for scripts

This commit is contained in:
VictorSanh
2019-09-11 14:19:07 +00:00
parent b62abe87c9
commit 32e1332acf
2 changed files with 11 additions and 3 deletions

View File

@@ -21,8 +21,12 @@ import random
import time
import numpy as np
from pytorch_transformers import BertTokenizer
import logging
from examples.distillation.utils import logger
logging.basicConfig(format = '%(asctime)s - %(levelname)s - %(name)s - %(message)s',
datefmt = '%m/%d/%Y %H:%M:%S',
level = logging.INFO)
logger = logging.getLogger(__name__)
def main():
parser = argparse.ArgumentParser(description="Preprocess the data to avoid re-doing it several times by (tokenization + token_to_ids).")
@@ -74,4 +78,4 @@ def main():
if __name__ == "__main__":
main()
main()