TorchFly is a PyTorch Fast Development Kit. The purpose is to learn the pipelines of the SOTA algorithms in Deep Learning areas like CV, NLP and RL. The utilities provided in this kit will shorten the time needed to rebuild some basic functions. Now, the kit is mainly for personal use, but will be updated from time to time.
You would need apex
Installing is simple using pip.
pip install torchflyYou can clone the repository. ```bash
git clone https://github.com/qywu/TorchFly.git
cd torchfly
pip install -e .
```
It is recommended to run on Nvidia Docker for better performance.
```bash
FROM nvcr.io/nvidia/pytorch:18.12.1-py3
RUN apt-get update
RUN pip install torchfly
```
-
Custom Bucket Sampler
-
Custom Beam Search