FAQ6.md

6.如何将训练的记录保存成txt

①在 ./tools/train.py 中写入 write() 函数

def write(logstats):
    filename = 'log.txt'
    with open(filename,'a',encoding='utf-8') as f:
        f.writelines(logstats + '\n')
        f.close

② 找到此段代码，加入 write(strs) 代码

if it % cfg.log_iter == 0 and (not FLAGS.dist or trainer_id == 0):
    strs = 'iter: {}, lr: {:.6f}, {}, time: {:.3f}, eta: {}'.format(
        it, np.mean(outs[-1]), logs, time_cost, eta)
    logger.info(strs)
    write(strs)