Hi. I am also creating a list of batch which include dict.
I am trying to use nn.DataParallel but also got error, how did you solve the problem?
RuntimeError: Caught RuntimeError in replica 0 on device 0.