Skip to content

v1.0.0

Latest
Compare
Choose a tag to compare
@GabrielDornelles GabrielDornelles released this 29 Oct 22:28
· 9 commits to main since this release

In this release is provided the trained weights.
I've trained every model for 100 epoches with learning rating 3e-4

  • Yolov1 weights is pth file, so it only contains a state_dict.
  • pix2pix_generator is a tar dictionary, when loaded, the state_dict is inside the key "state_dict".
  • crnn_and_vocab.pth.tar is a dictionary with the state_dict in "state_dict" key and the vocabulary class in "vocabulary" key.