I defined a custom model and a custom environment (I’m using Pytorch).
It seems my model is first called with some dummy input (I put print statements in both my environment and model, but the print of the model appears first).
The problem is that these dummy inputs crashes my model (mask with all values to 0).
Hey @Astariul , yeah, this is the initial test pass when initializing the loss. Contains all 0s.
We are not intending to change this, though, but could you try to catch this in your model for now with some if block?