model_runner.py 文件源码

def plot_losses(conf,losses_list,builder,name=''):
    unique_id = builder.get_unique_id()
    savedir = 'losses'
    if not os.path.exists(savedir):
        os.makedirs(savedir)

    save_path = os.path.join(savedir,'{}_loss_{}.png'.format(name,unique_id))
    pl.figure()
    for losses in losses_list:
        pl.semilogy(losses)
    pl.xlabel('Epoch')
    pl.ylabel('Loss')
    pl.grid()
    pl.savefig(save_path)