wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change min max gpu memory to be on their own plots (#1358)

Browse Source
This commit is contained in:
Tullie Murrell authored and GitHub committed 2020-04-03 14:59:27 -04:00
1 parent 003ca510fa
commit b31edf37bf
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/core/memory.py
+1 -1
View File
@@ -246,7 +246,7 @@ def get_memory_profile(mode: str) -> Union[Dict[str, int], Dict[int, int]]:
min_index, min_memory = min(memory_map.items(), key=lambda item: item[1])
max_index, max_memory = max(memory_map.items(), key=lambda item: item[1])
memory_map = {min_index: min_memory, max_index: max_memory}
memory_map = {'min_gpu_mem': min_memory, 'max_gpu_mem': max_memory}
return memory_map