训练集、验证集区分问题
chuanleiD opened this issue · comments
up主你好,你的工作对我很有帮助。在finetune_webui.py代码中,您似乎使得训练集与验证集相同,请问这种方法是否合适?是否在训练中会出现过拟合的情况?非常感谢
final_annos = cleaned_new_annos
# save annotation file
with open("final_annotation_train.txt", 'w', encoding='utf-8') as f:
for line in final_annos:
f.write(line)
# save annotation file for validation
with open("final_annotation_val.txt", 'w', encoding='utf-8') as f:
for line in cleaned_new_annos:
f.write(line)