search

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.
MIT License
98 stars 8 forks source link

Examples on using this library with QAT #5

Closed steven-spec closed 4 months ago

steven-spec commented 8 months ago

Thank you for such an easy library to use. My question is, do you have some examples as to how to use Quantized Aware Training with this library? Is there something specific I would need to know to be able to train with this library in QAT mode?

isaaccorley commented 8 months ago

Hi @steven-spec I'm currently working on adding an example notebook which uses PyTorch Lightning. I believe it supports QAT. See more information here.

Let me know if this answers your question.

notprime commented 5 months ago

Hi @steven-spec , does Isaac's answer suit your need? So that we can close this issue. Btw, we'll provide lots of tutorial for torchseg in the near future 😎

steven-spec commented 4 months ago

Hi @steven-spec , does Isaac's answer suit your need? So that we can close this issue. Btw, we'll provide lots of tutorial for torchseg in the near future 😎

yes, thanks, this issue can close