wgmueller1 / ulm-basenet

ULMFit implementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

conda install pytorch pytorch=0.3.1 cuda90 -c pytorch -y
pip install git+https://github.com/bkj/basenet
conda install -c conda-forge spacy -y
python -m spacy download en

pip install -r requirements.txt

Usage

See ./run.sh for usage.

Todo

  • Update to pytorch==0.4 + test
  • Test installation

About

ULMFit implementation


Languages

Language:Python 93.8%Language:Shell 6.2%