I want to log the training and inference process of a PyTorch neural network, including the power consumption with jtop. The limitation of using wandb.ai is that it requires either (a) network connection or (b) a self hosted docker container both of which disturb the power measurements.
Any recommendations on how I could go about this?