/pytorch_template

PyTorch Template makes deep learning project easier!

Primary LanguagePythonMIT LicenseMIT

PyTorch Template

pytorch template is to make building a deep learning model better and faster.

project directory

  • datas ：datas file
  • files ：files
  • src
    • data_loader ： data loaders
    • model ：models
    • loss ：loss functions
    • metric ：evaluation metrics
    • myutils ：some utils
    • trainer ：model trainer
    • config.json ：config file
    • train.py ：train file(run)
    • test.py ：test file(run)
  • readme.md ：project instructions file

Develop Requirements

  • OS: Ubuntu 16.04 LTS (64bit)
  • GPU: Nvidia Titan Xp
  • Language: Python 3.6.2.
  • Pytorch: 0.4.1
  • nltk==3.2.4
  • tensorboardX==0.8
  • torchtext==0.3.1
  • jieba==0.39
  • spacy==2.0.11
  • gensim==3.5.0

reference resources

[1] Pytorch NLP Template Project

[2] PyTorch Template Project