Skip to content
Snippets Groups Projects
Commit c8d7c137 authored by Steve Schmerler's avatar Steve Schmerler
Browse files

extract_model_params(): try_item=False by default

parent 556ac218
Branches
Tags
1 merge request!2Update GP slides and notebooks
...@@ -2,7 +2,7 @@ from matplotlib import pyplot as plt ...@@ -2,7 +2,7 @@ from matplotlib import pyplot as plt
import torch 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. """Helper to convert model.named_parameters() to dict.
With raw=True, use With raw=True, use
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment