Could I specify my own models or only the mentioned models only.

tomgoldstein / loss-landscape

Code for visualizing the loss landscape of neural nets

MIT License

2.72k stars 388 forks source link

Open alqurri77 opened 4 months ago

alqurri77 commented 4 months ago

I wonder if this implementation is for the models that is mentioned or if I can specify my own model.

tirtho109 commented 3 months ago

I have the same question. It would be nice to get some feedback.