资源算法allennlp_visdom

allennlp_visdom

2020-04-08 | |  37 |   0 |   0

我想使用visdom功能实时监测

AllenNLP

因此魔改一下allennlp,添加visdom的功能。

注意,使用json文件配置的话就不能使用这个功能,因为我还是初学者,有些函数还并不了解。


  • 仅仅更改的是train文件。

    • 有趣的是,tensorboard可能会报一个错,是多传入了epoch=epoch+=1,
      不清楚是不是因为fork的代码是旧版的代码(?)

使用方法是传入一个定义好的visdom的对象,
visdom_object=vis.Visdom(port=9010, env="BERT_ALLEN")

trainer = Trainer(
    model=model,
    optimizer=optimizer,
    iterator=iterator,
    train_dataset=train_ds,
    cuda_device=0 if USE_GPU else -1,
    num_epochs=config.epochs,
    validation_dataset=val_ds,
    patience=config.patience,
serialization_dir='./bert_saved',
num_serialized_models_to_keep=5,
    visdom_object=visdom_object
)

效果如图所示,实时监察运行情况。

图片.png

注意,因为我是用epoch作为横轴,
allennlp默认是断点继续训练,因此会有图像从非零开始,这说明你之前训练的模型没有删除。
如果一定要从坐标原点开始,建议直接删除已保存的文件。




上一篇:allennlp_source_code_analysis

下一篇:allennlp_sandbox

用户评价
全部评价

热门资源

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • shih-styletransfer

    shih-styletransfer Code from Style Transfer ...