pytorch에 있는 Transformer Layer 연습
- torch == 1.12.0+cu116
- datasets == 2.3.2
- tqdm == 4.62.3
- nltk == 3.6.5
- attn_heads: 4
- epoch: 100
- batch_size: 32
- hidden_size: 64
- max_len: 200
- learning_rate: 0.001
-
Notifications
You must be signed in to change notification settings - Fork 0
BWAAEEEK/IMDB_Transformer_pytorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published