input 썸네일형 리스트형 [Pytorch] Inputting specified model parameters to optimizer params_to_optimize = [param for name, param in self.model.named_parameters() if self.check_name_validity(TRAIN_WEIGHT_LIST, name)]self.opt = optim.AdamW(params_to_optimize, lr=config['lr'], weight_decay=config['w_decay']) 더보기 이전 1 다음