grad_scaler.scale(loss).backward

[lixinhaiyk]

taichi-nerfs/train.py

Line 180 in 3752d03

grad_scaler.scale(loss).backward()

hi,thanks for your amazing work,however,when I run the line grad_scaler.scale(loss).backward() ,it stop without any prompt in debug env.
In the command line env, it says"./scripts/train_nsvf_lego.sh: line 12: 3368608 Segmentation fault".

[Linyou]

Hi @lixinhaiyk! Could you please share your environment information, such as the version of taichi and GPU? Please note that the current version of taichi-nerfs requires taichi-1.6, so ensure that you have upgraded your taichi.