SSL4EO: convert script from MiB to MB (#1669)

This commit is contained in:
Adam J. Stewart 2023-10-20 10:05:18 -05:00 коммит произвёл Nils Lehmann
Родитель a6a11fd5b8
Коммит 5dba70014e
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -22,7 +22,7 @@ for model in models:
# Calculate memory requirements of model
mem_params = sum([p.nelement() * p.element_size() for p in m.parameters()])
mem_bufs = sum([b.nelement() * b.element_size() for b in m.buffers()])
mem = (mem_params + mem_bufs) / 2**20
mem = (mem_params + mem_bufs) / 1000000
print(f"Memory: {mem:.2f} MB")
with get_accelerator().device(0):