RuntimeError: triangles.is_contiguous() INTERNAL ASSERT FAILED at "rasterize_triangles.cpp"

[iamchenxin-coder]

This error occurs when I run the train.py script. I don't know how to fix it.
python train.py --data_dir $video_dir --num_epoch 20 --serial_batches False --display_freq 400 --print_freq 400 --batch_size 5

[RylanLiu-buaa]

之前用pytorch1.2和NVIDIA 1070就能通跑20s数据量的，但跑作者给的240s的就会显存不够。所以换成NVIDIA 3090，但需要pytorch1.8 这3090才能工作，但就会出上面这个triangles的错误。pytorch1.2 3090卡是不工作的直接程序就卡死在那了。估计就是pytorch的版本问题。

[iamchenxin-coder]

我使用Tesla V100的torch=1.8还是出现这个问题

[quqixun]

在 renderer/rasterizer.py 文件第19行前加入以下两行：

if not triangles.is_contiguous():
    triangles = triangles.contiguous()

测试环境为 Pytorch 1.8 + CUDA 11.1。

[xinwen-cs]

@quqixun thanks for PR
seems solved