Skip to content

ECS-251-W2020/final-project-torchfly

Repository files navigation


Notice

This is a repo under construction

Installment

Please download conda and create a virtual environment first.

# create virtual env
conda create -n torchfly python=3.6

apex is required, but it may need modifications if cuda version is mismatched.

# make sure ``nvcc`` is installed
# modified the error due to cuda version
git clone https://github.com/qywu/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

Then install this library in developer mode:

# clone repo
git clone https://github.com/ECS-251-W2020/final-project-torchfly

# install the repo
pip install -e .

To use the repo,

import torchfly_dev

Usage

Please see examples of how to use this repo:

  1. First preprocess your data into sectors. You should read this to do it. here

  2. Set up Google Cloud Preemptible Instance with instructions.

  3. Start training BERT with code here.

Weekly Meeting Notes

The most recent meeting notes are updated under meeting-notes directory.

Trello

The most recent progress and updates, https://trello.com/b/z7mDxem8/torchfly

About

PyTorch Deep Learning Toolkit

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published