Imagenet short tests (#6136)
Imagenet short tests (#6136)

* Add short imagenet tests (taken from seemuch)
- also rename to match go forward naming

* fix method name

* Update doc strings.

* Fixe gpu number.

* Add fill_objects

* fixed calling wrong class in super.

* fix lint issue.

derekjchow/models,derekjchow/models,alexgorban/models,tombstone/models,tombstone/models,tombstone/models,alexgorban/models,derekjchow/models,alexgorban/models,tombstone/models,derekjchow/models,alexgorban/models,tombstone/models,derekjchow/models,tombstone/models,alexgorban/models