Model Visualization

[Atmoboran]

Is there any way to visualize the model? I have mainly used tensorflow so far and i really liked the .plot_model function to get a feeling for what is happening inside the models. I have tried to use torchviz.make_dot on the Prithvi Models, but haven't managed to plot it. Does someone have a way to visualize it? (Not conceptually like in the paper, but actually showing all the layers and their connections.)
Best,
atmoboran

[Joao-L-S-Almeida]

I also didn't find a good way to visualize the models. When debugging I usually simply print them. As they are torch.nn.Module objects they will be displayed in a pretty-printed way, which could somehow work.

[Joao-L-S-Almeida]

Hi, @Atmoboran. I'm trying to use the approach available in the branch https://github.com/IBM/terratorch/tree/visualize/models. Check the notebook examples/notebooks/Visualize_model.ipynb.

[Atmoboran]

Hi @Joao-L-S-Almeida,
thanks that looks nice and can be very helpful.
Unfortunately I'm not able to run it, because somehow I can't install torchview and graphviz. I've seen quite a few posts on the internet that are struggling with installing it. Do you have the ultimate tip on how to do it? I have tried some ways, but couldn't make it work :/

[Joao-L-S-Almeida]

Strange, I have installed them inside a pyenv environment with python==3.11 without issues:

❯ pip install graphviz torchview
Collecting graphviz
  Downloading graphviz-0.20.3-py3-none-any.whl.metadata (12 kB)
Collecting torchview
  Downloading torchview-0.2.6-py3-none-any.whl.metadata (12 kB)
Downloading graphviz-0.20.3-py3-none-any.whl (47 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 47.1/47.1 kB 3.1 MB/s eta 0:00:00
Downloading torchview-0.2.6-py3-none-any.whl (25 kB)
Installing collected packages: torchview, graphviz
Successfully installed graphviz-0.20.3 torchview-0.2.6

[notice] A new release of pip is available: 24.0 -> 25.0.1
[notice] To update, run: python -m pip install --upgrade pip

[Joao-L-S-Almeida]

Are you using another environment manager ?Which errors you have seen ?