def set_parameter_requires_grad(model, feature_extracting):
if feature_extracting:
for param in model.parameters():
param.requires_grad = False
Why use only require_grad = False?
def set_parameter_requires_grad(model, feature_extracting):
if feature_extracting:
for param in model.parameters():
param.requires_grad = False
Why use only require_grad = False?
Does detach automatically change to require_grad = False?
Detach does not change requires grad. Detach is a tensor’s method which breaks model’s graph when used.
Thank you very much !!