Solved by Rakshit_Kothari in post #2 
              
              
                Responding to anyone who faces this. The fact that a tensor is wrapped in a dictionary is not relevant. DDP requires that all output variables be used in the graph. Calling ‘unused_parameters’ option did pretty much nothing.