pytorch-lightning注意事项
1、如果在代码中写了self.log(your_metric_dict, on_steps=True, on_epoch=True)那么monitor要写成monitor='val_loss_epoch'要不然会根据最后一步的验证尺度保存模型,不会根据整个周期的验证尺度保存模型,切记!!!!!!!!!!!!!!2、这种形式可以加载多个信息values = {'val_loss': loss, 'val_cer': cer_avg}self.log_dict(values, logger=T
原创
2020-12-12 10:38:03 ·
2050 阅读 ·
0 评论