Update freezing encoder mem engagement

computational-cell-analytics · Dec 22, 2024 · d8df10b · d8df10b
1 parent be3bdb6
commit d8df10b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/finetuning/livecell_finetuning.py b/finetuning/livecell_finetuning.py
@@ -58,7 +58,7 @@ def finetune_livecell(args):
 
     # NOTE: memory req. for all vit_b models (compared on A100 80GB)
     # vit_b
-    # freeze_encoder: ~ GB
+    # freeze_encoder: ~ 33.89 GB
     # QLoRA: ~48.54 GB
     # LoRA: ~48.62 GB
     # FFT: ~49.56 GB