UserWarning: Initializing zero-element tensors is a no-op...
This message is shown repeatedly at the starting of the training. Post that the training starts. At the end of the training, it fails with a text memory error and no outputs are saved. In wandb, multiple instan