Rename public-facing CodonGPT strings to CodonTranslator
Browse files- src/trainer.py +2 -2
src/trainer.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# src/trainer.py
|
| 2 |
"""
|
| 3 |
-
FSDP trainer for
|
| 4 |
No frameworks, no sugar. The model computes its own loss.
|
| 5 |
|
| 6 |
Batch invariants:
|
|
@@ -339,7 +339,7 @@ class Trainer:
|
|
| 339 |
# Initialize Weights & Biases (rank0 only)
|
| 340 |
if self._is_main():
|
| 341 |
if not hasattr(self, "_wandb"):
|
| 342 |
-
proj = os.environ.get("WANDB_PROJECT", "
|
| 343 |
name = os.environ.get("WANDB_NAME")
|
| 344 |
run_id = os.environ.get("WANDB_RUN_ID")
|
| 345 |
resume = os.environ.get("WANDB_RESUME")
|
|
|
|
| 1 |
# src/trainer.py
|
| 2 |
"""
|
| 3 |
+
FSDP trainer for CodonTranslator.
|
| 4 |
No frameworks, no sugar. The model computes its own loss.
|
| 5 |
|
| 6 |
Batch invariants:
|
|
|
|
| 339 |
# Initialize Weights & Biases (rank0 only)
|
| 340 |
if self._is_main():
|
| 341 |
if not hasattr(self, "_wandb"):
|
| 342 |
+
proj = os.environ.get("WANDB_PROJECT", "codontranslator")
|
| 343 |
name = os.environ.get("WANDB_NAME")
|
| 344 |
run_id = os.environ.get("WANDB_RUN_ID")
|
| 345 |
resume = os.environ.get("WANDB_RESUME")
|