Skip to main content

Pytorch Fast Development Kit

Project description

TorchFly

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.

Installation

Installing via pip:

Installing is simple using pip.

pip install torchfly

Installing from source

You can clone the repository. bash git clone https://github.com/qywu/TorchFly.git

Docker

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
```

TODOS

  1. Custom Bucket Sampler

  2. Custom Beam Search

Code References

FastAI

AllenNLP

Pytorch BERT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

TorchFly-0.0.1.tar.gz (6.0 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page