- 博客(2)
- 收藏
- 关注
原创 pytorch实践-bug/报错记录
1. RuntimeError: CUDA error: device-side assert triggered原因: 通常是由数组越界造成!仔细检查就能发现注意: 一般情况,程序报错的代码和真正的出错代码不是同一行解决办法:代码加上:import osos.environ['CUDA_LAUNCH_BLOCKING'] = "0"转自https://blog.csdn.net/Hungryof/article/details/97248209...
2020-12-25 13:36:44
219
原创 Keras实践-取一个模型中的子模型(模型分割)
问题由于经常调用各种预训练模型,而这些模型又常常封装得太好,不好修改,一种办法是取出所需要的子模型作为层再进行拼接设计。当我采取如下方法得到子模型(基于keras-bert)后出错:#加载模型bert_model = load_trained_model_from_checkpoint(self.config_path, self.checkpoint_path)#在Embedding-...
2019-10-25 21:48:04
1955
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人