Hi,
it seems your network doesn’t backprop properly.
you can check where by doing
for name,param in model.named_parameters():
if param is None:
   print(name)

So if it happens just after the loss (which is probably the case as I assume it’s been coded by yourself, you may find the guilty.