n0obcoder · November 21, 2019 18:08
diff --git a/mobile_gallery_pytorch_05_freeze.py b/mobile_gallery_pytorch_05_freeze.py
 #  We would freeze all but the last few layers (layer4 and fc)

 for name, param in model.named_parameters():
    if ('layer4' in name) or ('fc' in name):
        param.requires_grad = True
    else:
        param.requires_grad = False
	# We would freeze all but the last few layers (layer4 and fc)

	for name, param in model.named_parameters():
	if ('layer4' in name) or ('fc' in name):
	param.requires_grad = True
	else:
	param.requires_grad = False
No results found