demo.yaml 827 B

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