Increase compatibility with models with fully connected layers
Increase compatibility with models with fully connected layers

crowsonkb/style_transfer,crowsonkb/style_transfer,crowsonkb/style_transfer,crowsonkb/style_transfer