A feature request to add more control in the PyTorch logging util. is tracked here so you might want to check if this request would also cover your use case.
A feature request to add more control in the PyTorch logging util. is tracked here so you might want to check if this request would also cover your use case.