Correct LearningToPaint to match the wild (#560)
Summary:
Previously this model was only running the Stroked-Based
Renderer model, which is a simple nn with linear and conv
layers. But the real LearningToPaint model consists of
WGAN, discriminator and renderer. Adding the correct model
implementation.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/560
Reviewed By: xuzhao9
Differential Revision: D32400814
Pulled By: aaronenyeshi
fbshipit-source-id: 82bbfdc0f443cc3d3339f2756bb050693b216e7d