Spaces:

qgallouedec
/

train_memory

Running

qgallouedec HF Staff commited on Sep 4

Commit

7b17c8c

verified ·

1 Parent(s): c5962aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -103,7 +103,7 @@ with gr.Blocks() as demo:
                 with gr.Accordion("Hardware"):
                     num_gpus = gr.Number(1, label="Number of GPUs")
                 with gr.Accordion("Data"):
-                    grad_ckpt = gr.Checkbox(False, label="Gradient Checkpointing")
             submit = gr.Button("Submit")

                 with gr.Accordion("Hardware"):
                     num_gpus = gr.Number(1, label="Number of GPUs")
                 with gr.Accordion("Data"):
+                    grad_ckpt = gr.Checkbox(True, label="Gradient Checkpointing")
             submit = gr.Button("Submit")