search

logix-project / logix

AI Logging for Interpretability and Explainability🔬
Apache License 2.0
76 stars 6 forks source link

Hessian logging #30

Closed YoungseogChung closed 9 months ago

YoungseogChung commented 9 months ago

Finished implementing hessian logging.

In order to do this, I had to pass the "log_dir" information through global_config. The config sharing happens within config.py.

sangkeun00 commented 9 months ago

LGTM. Would you be also able to add a hessian loading logic? It would only require calling several torch.load() given the log directory.

sangkeun00 commented 9 months ago

LGTM!