How to inspect the whole Pytorch graph and save the training gif for kuka grasping?

[2000222]

Thanks a lot for your code. This is a solid and interesting project.

<1>I have run through your code in the jupyter notebook and I found you use "SummaryWriter" to inspect some scalars in the Tensorboard. Also I have tried the following code to inspect the whole PPO graph with "SummaryWriter" but I failed.

""" dummy_input = torch.rand(1,3,40,40).to(device) with SummaryWriter(log_dir = 'kuka-Diverse-ppo',comment='ActorCritic') as w: w.add_graph(policy_net,(dummy_input,)) """

I am not very familiar with the summarywriter, so could you give me more tutorials on this issue?

<2>Could you also teach us how to generate the training gif for pybullet-kuka-grasping as shown in your repo? I am so interested in it.
Thank you so much !

[mahyaret]

Thanks for your interest, here is a very good tutorial by pytorch: tensorboard tutorial

for generating gif in linux I use peek: sudo apt install peek in windows: screentogif