Commit 3f0a9b8d authored by wanglch's avatar wanglch
Browse files

Update single_dcu_train.py

parent dcb1010d
......@@ -202,7 +202,7 @@ if __name__=='__main__':
# 获取当前的日期和时间
now = datetime.now()
timestamp = now.strftime("%Y%m%d_%H%M%S")
new_model_path = f'saves/umt5_{timestamp}'
new_model_path = f'./saves/umt5_{timestamp}'
model.module.save_pretrained(new_model_path)
tokenizer.save_pretrained(new_model_path)
print("Done!")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment