I have a pretrained network; i just want part of that and want to remove the rest that is not useful (to save memory)
how should i do that?
lets say i load my pretrained net as:
    load_name = os.path.join('data/pretrained_model/model_pretrained.pth')
    checkpoint = torch.load(load_name)
    model.load_state_dict(checkpoint['model'], strict=False)

Can I delete the part of network that is not useful/used?