Spaces:
Runtime error
Runtime error
Commit
•
a981ff2
1
Parent(s):
96ce6e5
narrow layout
Browse files
app.py
CHANGED
@@ -44,12 +44,12 @@ memory = mem_calc.calculate_memory(args)
|
|
44 |
|
45 |
cols = st.columns(2)
|
46 |
cols[0].metric("GPU total", f"{memory['total_mem']:.2f} GB")
|
47 |
-
cols[1].metric("Offloaded
|
48 |
|
49 |
cols = st.columns(2)
|
50 |
-
cols[0].metric("Parameters
|
51 |
-
cols[1].metric("Activations
|
52 |
|
53 |
cols = st.columns(2)
|
54 |
-
cols[0].metric("
|
55 |
cols[1].metric(f"Optimizer ({'GPU' if offload else 'CPU'})", f"{memory['cpu_mem'] if offload else memory['optim']:.2f} GB", f"{memory['optim']/memory['total_mem'] * 100:.2f} %", delta_color="off")
|
|
|
44 |
|
45 |
cols = st.columns(2)
|
46 |
cols[0].metric("GPU total", f"{memory['total_mem']:.2f} GB")
|
47 |
+
cols[1].metric("Offloaded (RAM)", f"{memory['cpu_mem']:.2f} GB")
|
48 |
|
49 |
cols = st.columns(2)
|
50 |
+
cols[0].metric("Parameters", f"{memory['model']:.2f} GB", f"{memory['model']/memory['total_mem'] * 100:.2f} %", delta_color="off")
|
51 |
+
cols[1].metric("Activations", f"{memory['grad']:.2f} GB", f"{memory['grad']/memory['total_mem'] * 100:.2f} %", delta_color="off")
|
52 |
|
53 |
cols = st.columns(2)
|
54 |
+
cols[0].metric("CPU-GPU Transfer", f"{memory['overhead'] * 1000:.2f} ms")
|
55 |
cols[1].metric(f"Optimizer ({'GPU' if offload else 'CPU'})", f"{memory['cpu_mem'] if offload else memory['optim']:.2f} GB", f"{memory['optim']/memory['total_mem'] * 100:.2f} %", delta_color="off")
|