代码片段记录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) #使用