Question about using another model

Question

Question about using another model

Closed this issue 2 years ago · 2 comments

I want to train resnet model instead of mlp one.
How do I specify which model to use?

Answer 1 · 2022-04-18T18:50:14.000Z

You can change the name of the model in run_mpi.py for the specific dataset to resnet. Then you need to update the model_params dictionary with the appropriate ResNet parameters you need.

Answer 2 · 2022-04-19T13:28:32.000Z

Thank you for quick reply.
I tried what you said and it worked , I can train resnet model.
Thank you!!!