Port nesta's code into something readable, please.
Create a script experiments/plot_hessian.py that takes as inputs the dataset, model and training parameters (in the form of config.yaml or cmd arguments) downloads the artifacts from wandb, creates plot and uploads to wandb
Description
Port nesta's code into something readable, please.
Create a script
experiments/plot_hessian.py
that takes as inputs the dataset, model and training parameters (in the form of config.yaml or cmd arguments) downloads the artifacts from wandb, creates plot and uploads to wandb