How to use forward on modules in a ModuleList?

[wongkenchiang97 (Dylan)]

Then I used the Unet2D as a model for a training phase.
I’ve encounter “aten::resize is not implemented” runtime error during the backward of the loss at line 134.

Encounter "aten::resize is not implemented" runtime error during backward of loss

I’m using C++ frontend API to implement the training loop. “pred” are the the predictions of a 2D Unet model from a batch of training data. “target” are the label of the training data. “loss_f” is the torch::nn::BCEWithLogitsLoss(). “aten::resize is not implemented” runtime error appear at line 134. [image] Any idea to solve this issue?

Does this runtime error relevent to the forward style?