corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Trainer last summary fix + memory utils

Browse source
This commit is contained in:
Corentin 2021-02-25 02:18:02 +09:00
commit 50c395a07f
3 changed files with 54 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

19
utils/memory.py
View file

@ -6,3 +6,22 @@ def human_size(byte_count: int) -> str:
break
amount /= 1024.0
return f'{amount:.2f}{unit}B'
def human_to_bytes(text: str) -> float:
split_index = 0
while '0' <= text[split_index] <= '9':
split_index += 1
if split_index == len(text):
return float(text)
amount = float(text[:split_index])
unit = text[split_index:].strip()
if not unit:
return amount
if unit not in ['KiB', 'MiB', 'GiB', 'TiB', 'PiB', 'EiB', 'ZiB', 'YiB']:
raise RuntimeError(f'Unrecognized unit : {unit}')
for final_unit in ['KiB', 'MiB', 'GiB', 'TiB', 'PiB', 'EiB', 'ZiB', 'YiB']:
amount *= 1024.0
if unit == final_unit:
return amount