程序崩溃

[fufuwudi]

作者您好，尝试跑了一下您的开源代码，但是会出现非法内存的问题，想问下您遇到过这个问题么？

[fufuwudi]

应该是自定义算子tile_localagg_prob_sq的反向传播有问题：

1. 我把自定义算子注释掉（不使用自定义算子），可以正常跑通；
2. 我使用自定义算子，但是不做loss监督，通过会遇到该问题

[Yzichen]

不好意思，我这边没有遇到过这个问题。您具体的环境和GPU是什么呀？您的batch_size跟我的是对齐的吗？

[fufuwudi]

我的环境是python==3.8.18、torch==1.10.2+cu111、mmcv-full==1.4.0、mmdet=2.14.0、mmdet3d=0.17.2、mmsegmentation=0.14.1
配置的话，我使用的是projects/configs/superocc/superocc-t_r50_704_seq_nui_48e.py （batch size设置为1， num_frames=1）
配置这样设置会有影响么？

[Yzichen]

我建议先尝试换成我readme上推荐的环境。另外,这个错误是训练多少iters会出现呢? 以及你是用几张卡来训练的，我可以尝试复现一下。

[fufuwudi]

我这边用您官方的环境配置，是可以正常跑的（我推测可能是python版本不同导致的）
这个错误会在第一次迭代就会出现（我这边用4090单卡及8卡都能复现该问题），如果您有时间可以尝试复现一下
（您这边方便公布下训练log么，我想做一下loss对比）