IBM / mi-prometheus

Enabling reproducible Machine Learning research
http://mi-prometheus.rtfd.io/
Apache License 2.0
42 stars 18 forks source link

Change the way --model is handled #40

Closed vmarois closed 5 years ago

vmarois commented 5 years ago

As of now, the trainers have the option to load a pretrained model using the flag --model. We haven't really used this feature so far, and the way it is implemented (as a flag) makes it hard to handle in the grid trainers, which results in not handling this for now.

So I like to handle this in the grid trainers, because it useful to load a pretrained model for each experiment (e.g. to finetune a pre-trained model). I'm thinking of either:

We can discuss that :slightly_smiling_face:

vmarois commented 5 years ago

Adressed in #42