зеркало из https://github.com/microsoft/torchgeo.git
SSL4EO: convert script from MiB to MB (#1669)
This commit is contained in:
Родитель
a6a11fd5b8
Коммит
5dba70014e
|
@ -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):
|
||||
|
|
Загрузка…
Ссылка в новой задаче