files:
  model_file: trained.pth
python: 3.10.11
torch: 2.1.2
yaml_files:
  train_config: config.yaml