Tume233的博客

在使用CrossEntropyLoss()函数时，引发上述错误，大致意思是数据类型不匹配，CrossEntropyLoss()要求输入数据应为long型，但是我的数据必须是float类型的，所以只能改用MSELoss()函数，保持两个输入向量size一致即可。...

没有activate相关文件，无法激活虚拟环境解决方案：添加进去export PATH="~/miniconda3/bin:$PATH"

今天调试代码的时候发现一个错误：Missing key(s) in state_dict:错误代码：model.bert.load_state_dict(torch.load(args.init_checkpoint, map_location="cpu"))解决方案：model.load_state_dict(torch.load(args.init_checkpoint), False)...

服务器上跑程序，出现这种错误，大概是因为程序太大内存不够，有以下两种解决方式：1、batch_size设置太大，改小一点2、cuda节点被占用，可以指定其他节点，一种方式：程序里面写device = torch.device(‘cuda:0’ if torch.cuda.is_available() else ‘cpu’)，服务器里面写CUDA_VISIBLE_DEVICES=X；另一种方式：程序里面直接写os.environ[“CUDA_VISIBLE_DEVICES”] = “2,3”...

terminate called after throwing an instance of ‘std::runtime_error’运行pytorch程序报错：terminate called after throwing an instance of ‘std::runtime_error’what(): NCCL Error 1: unhandled cuda error解决方法：我之前下载的pytorch版本是1.4的，后来换成1.0.1的就好了...