search

tomgoldstein / loss-landscape

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

Visualizing a custom model #42

Open Avv22 opened 1 year ago

Avv22 commented 1 year ago

Hello,

Thank you for the tool. I have trained a Siamese network and stored the weights in an .h5 file after I finished training:

model.save('siamese_h5_model.h5')

Next, I am trying to plot both a 2D and a 3D plot for the network. Can you please help me how to do that?

KatarinaYuan commented 1 year ago

Hi, I have the same question here. Do you find a way to solve that? @Avv22 Thank you!

alqurri77 commented 4 months ago

Hi .. I also have the same issue.