有人能帮我解释一下Pytorch中这个常见问题的含义吗? 型号: EfficientDet-D4 GPU: RTX 2080Ti 批量:2 CUDA out of memory.11.00 GiB total capacity; 8.32 GiB already allocated; 2.59 MiB free; 8.37 GiB reserved in total by PyTorch) 无论如何,我认为模型和图形处理器在这里并不重要,我知道解决方案应该是减少批量大小,尝试关闭梯度,同时验证等。另外:我
我有一个问题,如何重新加载不同批量大小的pytorch模型。在训练中,我的批处理大小是64,但在推断中,我希望批处理大小是1(逐个输入数据)。我得到了这个错误:
RuntimeError: size mismatch, m1: [37 x 1], m2: [37 x 64] at /Users/soumith/code/builder/wheel/pytorch-src在重新加载pytorch模型中,有没有解决