diff --git a/utils/generic_utils.py b/utils/generic_utils.py index 3cdf74bc..983797ba 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -193,7 +193,6 @@ def set_weight_decay(model, weight_decay, skip_list={"decoder.attention.v"}): if not param.requires_grad: continue if len(param.shape) == 1 or name in skip_list: - print(name) no_decay.append(param) else: decay.append(param)