How I can eval model after each epoch end?

fredzzhang / upt

[CVPR'22] Official PyTorch implementation for paper "Efficient Two-Stage Detection of Human–Object Interactions with a Novel Unary–Pairwise Transformer"

https://fredzzhang.com/unary-pairwise-transformers

BSD 3-Clause "New" or "Revised" License

144 stars 26 forks source link

How I can eval model after each epoch end? #47

Closed SherlockHolmes221 closed 2 years ago

SherlockHolmes221 commented 2 years ago

How I can eval model after each epoch end? Thanks

fredzzhang commented 2 years ago

Hi @SherlockHolmes221,

The evaluation code is implemented for a single GPU, without distributed data parallel. So, it wasn't included during training. However, all checkpoints after each epoch will be cached by default in the directory you specified. Each checkpoint is named as ckpt_steps_epochs. You can run evaluation on the checkpoints you are interested in.

Fred.

SherlockHolmes221 commented 2 years ago

Thx