Fix model layer for resnet v1. (#4230)
Fix model layer for resnet v1. (#4230)

The final BN and ReLU layer is only need for v2 model since it was
doing preactivation in each block.
derekjchow/models,derekjchow/models,derekjchow/models,tombstone/models,cshallue/models,alexgorban/models,tombstone/models,tombstone/models,cshallue/models,cshallue/models,alexgorban/models,cshallue/models,tombstone/models,tombstone/models,derekjchow/models,alexgorban/models,cshallue/models,alexgorban/models,derekjchow/models,alexgorban/models,tombstone/models