Hello,
I’d like to visualize a preprainted bert model that’s very tiny. It inherits transformers/src/transformers/modeling_utils.py at eb5b968c5d80271ecb29917dffecc8f4c00247a8 · huggingface/transformers · GitHub
I tried using tensorboard but wasn’t successfull so far. I can get a summary but I’d like a bit more details. Any package capable of that?
Thanks in advance