Skip to content

mayokaze/ulm-basenet

 
 

Repository files navigation

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

About

ULMFit implementation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 93.4%
  • Shell 6.6%