I dont have an explicit backward pass, only the forward method of my model, or do you mean the backward invocation in my training loop?
Will the inclusion of this line at the beginning of my file then also apply to the crucial part in which I load in hdf5 files and convert them into tensors? Because this is where I suspect NaN values to occur…