schema.yaml

runtime:
  type: dict
  schema: 
    cuda: 
      type: boolean
      required: true
       nullable: false

logging:
  type: dict
  schema:
    wandb:
      type: dict
      schema:
        run_name:
          type: string
          required: true
          nullable: true
        project_name:
          type: string
          required: true
          nullable: true
    checkpoint:
      type: dict
      schema: 
        save_name: 
          type: string
          required: true
          nullable: true
        save_epochs:
          type: integer
          required: true
          nullable: true

training:
  type: dict
  schema:
    epochs: 
      type: integer
      required: true
      nullable: false
    batch_size: 
      type: integer
      required: true
      nullable: false
    subsample: 
      type: integer
      required: true
      nullable: frue