make resnet and mlp responsible to ensure embeddings are correct dims

make resnet and mlp responsible to ensure embeddings are correct dims #682