Model weights for training

[Gayathri142]

Hi, Could you please elaborate a bit on the parameter '--model_dir', in the training part of the tool? Are there weights that come with the study, or do we have to generate them ourselves? If so, could you please guide me on how to generate the weights and add it to the model while training? Thanks, Gayathri

[MicrobeLab]

Hi, The value of --model_dir sets the folder to store weights during training and the folder from which trained weights are loaded in prediction. So, the folder should be same for training and testing. In training mode, the model will store the trained weights to this folder.

[Gayathri142]

Thank you for your explanation.