Error with nested models "Caffe2 uses a lazy allocation..."

[Emanuele97x]

I am trying to run a FSDP training built on top of metaseq:

I have a model with two other models nested inside it:

• FSDP(model);
  • FSDP(submodel)
  • FSDP(submodel);

However I get this error: ""RuntimeError: The tensor has a non-zero number of elements, but its data is not allocated yet. Caffe2 uses a lazy allocation, so you will need to call mutable_data() or raw_mutable"" during the forward pass of the first submodule, do you ever encountered this error? Thanks a lot!