mnazari123

> > 我按照所写的说明进行了操作。我正在尝试在 1 个 GPU 中运行模型的分割部分；不幸的是，与深度不同，我遇到了一个无法解决的错误。如果您对使其适用于单 GPU 所需的更改有任何建议，我将不胜感激。 这是文件中的错误： 文件“/media/user/meta/segmentation/models/model.py”，第216行，在forward_train loss_decode = self._decode_head_forward_train(x, img_metas, File“/media/user/meta/分段/模型/model.py”，第 149 行，在 _decode_head_forward_train loss_decode = self.decode_head.forward_train(x, img_metas, > > 这是错误： 文件“/home/user/seg_env/lib/python3.10/site-packages/torch/nn/function.py”，第 2416 行，在 _verify_batch_size...