Can't load data parallel model

[NoamRosenberg]

When trying to load data parallel model I get an error which I can't see because the command line gets flooded with model layers like such. I thought this might have something to do with data parallel because of the "module", but even removing the "module" from the state_dict names or trying to load with model.module.load_state_dict doesn't work.

.
.
.
.
6.0.conv_p.1.running_var", "module.aspp_16.0.conv_p.1.num_batches_tracked", "module.aspp_16.0.concate_conv.0.weight", "module.aspp_16.0.concate_conv.1.weight", "module.aspp_16.0.concate_conv.1.bias", "module.aspp_16.0.concate_conv.1.running_mean", "module.aspp_16.0.concate_conv.1.running_var", "module.aspp_16.0.concate_conv.1.num_batches_tracked", "module.aspp_32.0.conv11.0.weight", "module.aspp_32.0.conv11.1.weight", "module.aspp_32.0.conv11.1.bias", "module.aspp_32.0.conv11.1.running_mean", "module.aspp_32.0.conv11.1.running_var", "module.aspp_32.0.conv11.1.num_batches_tracked", "module.aspp_32.0.conv33.0.weight", "module.aspp_32.0.conv33.1.weight", "module.aspp_32.0.conv33.1.bias", "module.aspp_32.0.conv33.1.running_mean", "module.aspp_32.0.conv33.1.running_var", "module.aspp_32.0.conv33.1.num_batches_tracked", "module.aspp_32.0.conv_p.0.weight", "module.aspp_32.0.conv_p.1.weight", "module.aspp_32.0.conv_p.1.bias", "module.aspp_32.0.conv_p.1.running_mean", "module.aspp_32.0.conv_p.1.running_var", "module.aspp_32.0.conv_p.1.num_batches_tracked", "module.aspp_32.0.concate_conv.0.weight", "module.aspp_32.0.concate_conv.1.weight", "module.aspp_32.0.concate_conv.1.bias", "module.aspp_32.0.concate_conv.1.running_mean", "module.aspp_32.0.concate_conv.1.running_var", "module.aspp_32.0.concate_conv.1.num_batches_tracked", "module.final_conv.weight", "module.final_conv.bias".

[NoamRosenberg]

I wrote CUDA_AVAILABLE_DEVICES instead of CUDA_VISIBLE_DEVICES. later I cleaned out the module wrapper and loaded it as I would a non data parallel model. For now this only works on a single GPU. However I still can't load parallel models on multi GPUs

[NoamRosenberg]

okay, for some reason in this version of pytorch we can load parallel model state_dict without specifically mentioning .module

I've committed and pushed a fix.