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

This commit is contained in:
Adam J. Stewart 2023-10-20 10:05:18 -05:00 коммит произвёл GitHub
Родитель 4126f464ab
Коммит 73791b6f1e
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
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):