Visual network architecture about alf HOT 6 CLOSED

Hi Thanks for the message.

The procedure to visualize the algorithm architecture of a conf file is as follows:

• intall graphviz: sudo apt-get install graphviz
• run the training for a few epochs
  python -m alf.bin.train --conf=path_to_conf --root_dir=path_to_results

The figure algorithm_sturcture.png will be saved to the root dir.

Let us know if you have further questions.

from alf.

sry... I didn't find anything like a png in root dir.

from alf.

oh, i find this "W0503 23:24:27.788165 140296919340224 policy_trainer.py:111] Need "graphviz" installed if you want to visualize modules".
But i have installed it.

from alf.

Hi @Mming11,

The graphviz module prompted by alf actually means the graphviz python module. Depending on how you build your python development environment for alf, you may need to do so with poetry add graphviz or pip install graphviz. The package you have installed via apt does not provide the python package, at least not directly in your development environment as it appears.

Also, what kind of visualization do you expect?

from alf.

Here is the visualization that I generated for ppg_metadrive_transformer. Hope it helps.

from alf.

Oh, that's great. Thank you very much!!!

from alf.