search

ServiceNow / embedding-propagation

Codebase for Embedding Propagation: Smoother Manifold for Few-Shot Classification. This is a ServiceNow Research project that was started at Element AI.
Apache License 2.0
208 stars 21 forks source link

Training #7

Closed cvprkang closed 2 years ago

cvprkang commented 4 years ago

I findthat the pre-training and fine-tuning time for this work is very long. Is there any way to speed up the training time?

prlz77 commented 4 years ago

You could try to use automatic mixed precision (amp) https://pytorch.org/blog/pytorch-1.6-released/