반응형
if config.model.use_param_regressor:
print(f"freeze all parameters except ParamRegressor")
for name, param in model.named_parameters():
if 'ParamRegressor' in name:
# print(f"{name} requires_grad = True")
pass
else:
param.requires_grad = False
# print(f"{name} requires_grad = False")
반응형
'연구 > pytorch' 카테고리의 다른 글
[Transformer] torch.nn.MultiheadAttention 모듈의 mask 인자 개념 (0) | 2023.01.20 |
---|---|
[npz] npz 데이터에서 keys 확인하기 (0) | 2022.07.19 |
[img 시각화] tensor로 변환된 이미지를 cv2.imshow 하기 (0) | 2021.12.20 |
[파이썬] 피클 pickle 파일 읽기, 쓰기 (0) | 2021.07.08 |
[error] 파이썬 data loader 시간지연문제 (1) | 2021.05.25 |