请问，如何定义主干网络返回两个特征图数组呢？head和loss如何写？

open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

https://mmsegmentation.readthedocs.io/en/main/

Apache License 2.0

7.68k stars 2.53k forks source link

Open stu0440 opened 3 weeks ago

stu0440 commented 3 weeks ago

报错信息：TypeError: MyHead.forward() missing 1 required positional argument: 'attentions'