BlinkDL commited on
Commit
8e885f3
1 Parent(s): 078ca32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -77,7 +77,8 @@ def evaluate(
77
  out_last = i + 1
78
 
79
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
80
- print(f'vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')
 
81
  del out
82
  del state
83
  gc.collect()
 
77
  out_last = i + 1
78
 
79
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
80
+ timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
81
+ print(f'{timestamp} - vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')
82
  del out
83
  del state
84
  gc.collect()