diff --git a/BLcourse2.3/utils.py b/BLcourse2.3/utils.py index c8a66c4cf4dd33fa5cf62d93e84287eabf33a405..0c0e605232bab6fd12cda350a2c612cad9c1b58b 100644 --- a/BLcourse2.3/utils.py +++ b/BLcourse2.3/utils.py @@ -2,7 +2,7 @@ from matplotlib import pyplot as plt import torch -def extract_model_params(model, raw=False, try_item=True) -> dict: +def extract_model_params(model, raw=False, try_item=False) -> dict: """Helper to convert model.named_parameters() to dict. With raw=True, use