alegendaryfish commited on
Commit
7c92c02
·
verified ·
1 Parent(s): 81f3272

Rename public-facing CodonGPT strings to CodonTranslator

Browse files
Files changed (1) hide show
  1. src/trainer.py +2 -2
src/trainer.py CHANGED
@@ -1,6 +1,6 @@
1
  # src/trainer.py
2
  """
3
- FSDP trainer for CodonGPT.
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", "codongpt")
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")