Add DataParallel meta-module in torchmeta.modules. This is identical to torch.nn.DataParallel, with full support for operations on MetaModule instances (pass additional params argument in forward(), access to meta_parameters()/meta_named_parameters() methods)
Add tests for DataParallel, conditional on multi-GPU support
DataParallel
meta-module intorchmeta.modules
. This is identical totorch.nn.DataParallel
, with full support for operations onMetaModule
instances (pass additionalparams
argument inforward()
, access tometa_parameters()
/meta_named_parameters()
methods)DataParallel
, conditional on multi-GPU support