forked from jthomme1/pytorch-original-transformer
-
Notifications
You must be signed in to change notification settings - Fork 4
/
environment.yml
23 lines (23 loc) · 756 Bytes
/
environment.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
name: rethinking-attention
channels:
- defaults
- pytorch
dependencies:
- python==3.8.3
- pip==20.0.2
- matplotlib==3.1.3
- pytorch==1.5.0
- torchtext==0.6.0
- numpy==1.20.3
- pip:
- GitPython==3.1.2
- jupyter==1.0.0
- numpy==1.20.3
- nltk==3.5
- seaborn==0.11.0
- spacy==2.3.2
- https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-2.3.1/en_core_web_sm-2.3.1.tar.gz#egg=en_core_web_sm
- https://github.com/explosion/spacy-models/releases/download/de_core_news_sm-2.3.0/de_core_news_sm-2.3.0.tar.gz#egg=de_core_news_sm
- https://github.com/explosion/spacy-models/releases/download/fr_core_news_sm-2.3.0/fr_core_news_sm-2.3.0.tar.gz#egg=fr_core_news_sm
- optuna
- datasets