how to modify the dropout rate of pretained model pytorch, for example mobilenetV2?
Thank you in advance.
You could directly access the attributes of the dropout layer, e.g. via:
model.classifier[0].p = 0.5
how to modify the dropout rate of pretained model pytorch, for example mobilenetV2?
Thank you in advance.
You could directly access the attributes of the dropout layer, e.g. via:
model.classifier[0].p = 0.5