n1-k0

修改batch问题

2

你好，我在修改Ocean.yaml的时候发现，将batch设为32或64能正常运行，如果继续调大（如128或192）则会出现如下错误 > `Traceback (most recent call last): File "./tracking/train_ocean.py", line 259, in main() File "./tracking/train_ocean.py", line 250, in main model, writer_dict = ocean_train(train_loader, model, optimizer, epoch + 1, curLR, config,...

你好，请问sa模块是否可以作为即插即用的子模块，用在特征图之后的其他任务分支上？

5

你好，请问sa模块是否可以作为即插即用的子模块，用在特征图之后的其他任务分支上？如使用resnet提取特征之后，特征图在不同head上的操作中加入sa模块