File size: 812 Bytes
1833b64 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | """Print the canonical Nomen-AI training configuration."""
from pathlib import Path
import yaml
def main():
cfg_path = Path(__file__).resolve().parents[1] / 'train_config.yaml'
with cfg_path.open() as f:
cfg = yaml.safe_load(f)
print('Project:', cfg['project'])
print('Base model:', cfg['base_model'])
print('SFT dataset:', cfg['artifacts']['sft_dataset'])
print('DPO dataset:', cfg['artifacts']['dpo_dataset'])
print('SFT output:', cfg['artifacts']['sft_adapter'])
print('DPO output:', cfg['artifacts']['dpo_adapter'])
print('SFT effective batch:', cfg['sft']['effective_batch_size'])
print('DPO effective batch:', cfg['dpo']['effective_batch_size'])
print('Implemented roots:', cfg['controls']['implemented_roots'])
if __name__ == '__main__':
main()
|