Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Some doubts about config.py #78

Open
dagongji10 opened this issue Jun 10, 2020 · 1 comment
Open

Some doubts about config.py #78

dagongji10 opened this issue Jun 10, 2020 · 1 comment

Comments

@dagongji10
Copy link

@zhang0jhon Thanks for sharing a nice work!
I plan to train my own datasets with AttentionOCR and when I modify config.py, I got some doubts:

  1. restore_path: how to use this para? Set it to /path/to/pretrained.ckpt? Need I change starting_epoch?
  2. steps_per_epoch: I only have 2 GPUs, so set batch_size=5, and how to change steps_per_epoch? My datasets have 56030 samples, must I set steps_per_epoch=56030/5?
  3. warmup_steps: how to use this para? Must it keep same with num_epochs?
@alan-img
Copy link

@zhang0jhon Thanks for sharing a nice work!
I plan to train my own datasets with AttentionOCR and when I modify config.py, I got some doubts:

  1. restore_path: how to use this para? Set it to /path/to/pretrained.ckpt? Need I change starting_epoch?
  2. steps_per_epoch: I only have 2 GPUs, so set batch_size=5, and how to change steps_per_epoch? My datasets have 56030 samples, must I set steps_per_epoch=56030/5?
  3. warmup_steps: how to use this para? Must it keep same with num_epochs?

your problem I don't know, but I want to know what is label tool of labeling you own datset. because I also want to train in my own dataset.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants