alegendaryfish commited on
Commit
53a2f0d
·
verified ·
1 Parent(s): a4b9658

Rename public-facing CodonGPT strings to CodonTranslator

Browse files
Files changed (1) hide show
  1. train.py +3 -3
train.py CHANGED
@@ -1,6 +1,6 @@
1
  #!/usr/bin/env python
2
  """
3
- Minimal, honest training script for CodonGPT on CSV data.
4
 
5
  - Species conditioning: REQUIRED (precomputed embeddings)
6
  - Protein conditioning (ESM-C): ENABLED BY DEFAULT. Disable with --no_protein.
@@ -21,7 +21,7 @@ logging.basicConfig(
21
  datefmt="%m/%d/%Y %H:%M:%S",
22
  level=logging.INFO,
23
  )
24
- logger = logging.getLogger("codongpt.train")
25
 
26
  def _describe_sdp_kernels() -> None:
27
  # Log the enabled SDPA backends (Flash/MemEff/Math) without raising on older PyTorch
@@ -57,7 +57,7 @@ def _speed_toggles():
57
 
58
 
59
  def parse_args():
60
- p = argparse.ArgumentParser(description="Train CodonGPT on CSV data")
61
  # Data (CSV path or Parquet glob/dir)
62
  p.add_argument("--train_data", type=str, default="random_sample_1000.csv",
63
  help="Training data: CSV file or Parquet glob/dir (e.g., ./data/train_shards/*.parquet)")
 
1
  #!/usr/bin/env python
2
  """
3
+ Minimal, honest training script for CodonTranslator training data.
4
 
5
  - Species conditioning: REQUIRED (precomputed embeddings)
6
  - Protein conditioning (ESM-C): ENABLED BY DEFAULT. Disable with --no_protein.
 
21
  datefmt="%m/%d/%Y %H:%M:%S",
22
  level=logging.INFO,
23
  )
24
+ logger = logging.getLogger("codontranslator.train")
25
 
26
  def _describe_sdp_kernels() -> None:
27
  # Log the enabled SDPA backends (Flash/MemEff/Math) without raising on older PyTorch
 
57
 
58
 
59
  def parse_args():
60
+ p = argparse.ArgumentParser(description="Train CodonTranslator on CSV or Parquet data")
61
  # Data (CSV path or Parquet glob/dir)
62
  p.add_argument("--train_data", type=str, default="random_sample_1000.csv",
63
  help="Training data: CSV file or Parquet glob/dir (e.g., ./data/train_shards/*.parquet)")