Hi DefeatedCrab47
If you are referring to this example, examples/frameworks/tensorboardx/pytorch_tensorboardX.py, it does have only test and train steps.
If you like to plot validation together with the train, you can have the same prefix, for example when using writer.add_scalar('<prefix>/Test_Loss', ...)
, like in this example - https://demoapp.trains.allegro.ai/projects/bb21d73db5634e6d837724366397b6e2/experiments/f46160152ee54ff9863bb2efe554a6b1/output/metrics/scalar
Ah I see, it's based on a naming scheme, thanks. Sorry I forgot to link the tutorial I was looking at: https://allegro.ai/docs/examples/frameworks/pytorch/pytorch_tensorboard/