I would like to monitor the change of weights in every epoch.
So i use torch.save.
Then, i train ResNet-34 for CIFAR-10 image classification.
If the above function is used, a large amount of tensors are recorded, making it difficult to measure.
So, I want to monitor the weight of a specific layer or measure the mean of a specific tensor.
Any advice would be appreciated!