How to generate the direction and surface file (.h5 files)?

tomgoldstein / loss-landscape

Code for visualizing the loss landscape of neural nets

MIT License

2.72k stars 388 forks source link

How to generate the direction and surface file (.h5 files)? #23

Closed DAIZHENWEI closed 5 years ago

DAIZHENWEI commented 5 years ago

I can implement plot the surface of the function using the trained networks attached. If I want to apply the code on a new model, could you provide more details about how to generate the .t7 and .h5 files?