search

tomgoldstein / loss-landscape

Code for visualizing the loss landscape of neural nets
MIT License
2.72k stars 388 forks source link

#custom dataset #40

Open Harry-KIT opened 1 year ago

Harry-KIT commented 1 year ago

Hi @ljk628 good job! Do you also have a plan to adopt it for custom data?