The replacement of .module points to loading a state_dict from a nn.DataParallel model and should most likely be unrelated to the error.
Do you see any values in your loaded state_dict for the LogSoftmax layer?