demo.yaml 807 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. model:
  2. arch: unimernet
  3. model_type: unimernet
  4. model_config:
  5. model_name: ./models
  6. max_seq_len: 1024
  7. length_aware: False
  8. load_pretrained: True
  9. pretrained: ./models/pytorch_model.bin
  10. tokenizer_config:
  11. path: ./models
  12. datasets:
  13. formula_rec_eval:
  14. vis_processor:
  15. eval:
  16. name: "formula_image_eval"
  17. image_size:
  18. - 192
  19. - 672
  20. run:
  21. runner: runner_iter
  22. task: unimernet_train
  23. batch_size_train: 64
  24. batch_size_eval: 64
  25. num_workers: 1
  26. iters_per_inner_epoch: 2000
  27. max_iters: 60000
  28. seed: 42
  29. output_dir: "../output/demo"
  30. evaluate: True
  31. test_splits: [ "eval" ]
  32. device: "cuda"
  33. world_size: 1
  34. dist_url: "env://"
  35. distributed: True
  36. distributed_type: ddp # or fsdp when train llm
  37. generate_cfg:
  38. temperature: 0.0