代码片段记录2

发表于 | 更新于:

本周主要看了AllenNLP/ELMO的代码，但并没有找到很多可复用的代码。本周也没有比较有意义的代码。

1️⃣get_time_diff

获取已使用的时间

1
2
3
4
5
6
7
8
9
import time
from datetime import timedelta

start_time=time.time()
def get_time_dif(start_time):
    """获取已使用时间"""
    end_time = time.time()
    time_dif = end_time - start_time
    return timedelta(seconds=int(round(time_dif)))

2️⃣parser使用

1
2
3
4
5
6
7
8
parser = argparse.ArgumentParser()
parser.add_argument('--save_dir', help='Location of checkpoint files')
parser.add_argument('--vocab_file', help='Vocabulary file')
parser.add_argument('--train_prefix', help='Prefix for train files')

args = parser.parse_args()
    
main(args)   #使用