maml_omniglot model with functorch (#1179)
Summary:
Very similar to the maml_omniglot model in torchbench (which is what we originally based this model off of). We also load some sample inputs from there because the model takes the same inputs.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1179
Test Plan: - python test.py -k TestBenchmark.test_functorch_maml_omniglot_train_cuda
Reviewed By: xuzhao9
Differential Revision: D39671689
Pulled By: zou3519
fbshipit-source-id: 2c1c6b92de5d32c0b37de5d069caa0f7bbb61e7d
