Both approaches should work and even nn.DataParallel should create chunks of your dict in case you want to unwrap it inside the forward method.
I don’t know of any advantages of one approach over the other, if you need to use dicts.