set whole model to requires_grad=False, this is for nerf model
(model)
| 50 | return feature |
| 51 | |
| 52 | def disable_model_grad(model): |
| 53 | ''' set whole model to requires_grad=False, this is for nerf model ''' |
| 54 | # print("disable_model_grad...") |
| 55 | for module in model.modules(): |
| 56 | # print("this is a layer:", module) |
| 57 | if hasattr(module, 'weight'): |
| 58 | module.weight.requires_grad_(False) |
| 59 | if hasattr(module, 'bias'): |
| 60 | module.bias.requires_grad_(False) |
| 61 | return model |
| 62 | |
| 63 | def inference_pose_regression(args, data, device, model, retFeature=False, isSingleStream=True, return_pose=True): |
| 64 | """ |
no outgoing calls
no test coverage detected