pytorch
mhuang..
这个作者很懒,什么都没留下…
展开
-
RuntimeError: Function AddmmBackward returned an invalid gradient at index 1 - got [4, 1024] but exp
今天做图像检测练习的时候报了这个错,反复调整了好几次参数都没用。后来反应过来,是因为我是直接加载已经训练过的模型,里面的参数数量已经是确定了的,而这个加载的模型和我现在操作的这个模型的输入大小不一样,所以不管怎么改模型内部的参数都没用。只能把输入大小改成之前那个训练过的模型的输入大小。...原创 2021-09-23 18:16:05 · 1311 阅读 · 0 评论 -
RuntimeError: CUDA out of memory 解决办法
目前看来两个问题应该都是由于显存不足导致解决方法一:关闭多余占用显存的进程 (我同时开了两个就报第二个错)解决方法二:减小batch_size 从最小的往上加或者在所有调用gpu的语句前后加上torch.cuda.empty_cache() 释放多余缓存例如:改为解决方法三:换显卡 或者找网上的云计算平台训练(例如google colab)...原创 2021-09-20 12:26:38 · 2412 阅读 · 0 评论 -
利用GPU训练模型出现:CUDA driver initialization failed
选择GPU训练时出现问题:CUDA未初始化if torch.cuda.is_available(): jy = jy.cuda() print("执行GPU")loss_fn = nn.CrossEntropyLoss()if torch.cuda.is_available(): loss_fn = loss_fn.cuda() print("执行gpu")可以试试更新显卡驱动:问题解决...原创 2021-09-20 11:10:31 · 6853 阅读 · 6 评论