Viewing a model

SameDiff
1

Hello, everyone

I saw, in this example (https://github.com/eclipse/deeplearning4j/blob/cbf1fad16c3fefc3009a3048ee840aa02a0c3d99/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/BERTGraphTest.java#L420), that it should be possible to visualize the model by writing it to a file (BERT_UI.bin) and using DL4J UI (UIServer?) for viewing.

Does anyone knows how? I couldn’t find any reference or example to this…

Cheers,

/rp

2

Found it…
Just add:

        UIServer uiServer = UIServer.getInstance();


        Thread.sleep(1000000000);

To the end of the function and execute.
Afterwards, point the browser to: localhost:9000/samediff, open the file and enjoy…

/rp