KeOps error when training the model

Dear author @abdulfatir, thanks for your great work!

I got this error during the evaluation step at epoch 49:

FileNotFoundError: [Errno 2] No such file or directory: '/home/ubuntu/.cache/keops2.1.1/build/tmp.txt'

I guessed this was related to pykeops so I upgraded pykeops to 2.2.3, but this time the training loss was always nan. Could you help me sovle this issue?

Bests