/ulm-basenet

ULMFit implementation

Primary LanguagePython

basenet-ulmfit

Implementation of ulmfit using basenet wrappers.

Code in ulmfit.py is adapted directly from fastai code.

Installation

conda create -n ulm_env python=3.6 pip -y
source activate ulm_env

# pytorch
conda install pytorch pytorch=0.3.1 cuda90 -c pytorch -y

# spacy (for tokenization)
conda install -c conda-forge spacy -y
python -m spacy download en

# additional requirements
pip install -r requirements.txt

Usage

See ./run.sh for usage.

Todo

  • Update to pytorch==0.4 + test