Open florian-huber opened 4 months ago
So far only a simply writer for use with Tensorboard.
writer.add_graph
So far only a simply writer for use with Tensorboard.
writer.add_graph
to also track the model architecture