Rename public-facing CodonGPT strings to CodonTranslator
Browse files
train.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
#!/usr/bin/env python
|
| 2 |
"""
|
| 3 |
-
Minimal, honest training script for
|
| 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("
|
| 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
|
| 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)")
|