10 lines
400 B
Python
10 lines
400 B
Python
from pytorch_lightning.callbacks.gpu_stats_monitor import GPUStatsMonitor
|
|
from pytorch_lightning.utilities import rank_zero_warn
|
|
|
|
|
|
class GpuUsageLogger(GPUStatsMonitor):
|
|
def __init__(self, *args, **kwargs):
|
|
rank_zero_warn("`GpuUsageLogger is now `GPUStatsMonitor`"
|
|
" and it will be removed in v0.11.0", DeprecationWarning)
|
|
super().__init__(*args, **kwargs)
|