Skip to content

Latest commit

 

History

History
22 lines (18 loc) · 1.12 KB

README.md

File metadata and controls

22 lines (18 loc) · 1.12 KB
Raw

SVMTransformer

This is a project of implementing SVM Transformer to do machine translation tasks, written in PyTorch. The theory is based on "Transformers as Support Vector Machines"

Changed the Attention Mechanism into SVM Attention, haven't reached a convergence due to the lack of computing resources, but from the current results, at least I could say that the new attention mechanism converges faster.

Usage

To use this just replace Transformer into SVMTransformer.

from models.model.transformer_svm import Transformer_SVM

TIPs

The link for Dataset Multi30k is broken, so I changed the urls inside torchtext into the following:

URL = {
    "train": r"https://raw.githubusercontent.com/neychev/small_DL_repo/master/datasets/Multi30k/training.tar.gz",
    "valid": r"https://raw.githubusercontent.com/neychev/small_DL_repo/master/datasets/Multi30k/validation.tar.gz",
    "test": r"https://raw.githubusercontent.com/neychev/small_DL_repo/master/datasets/Multi30k/mmt16_task1_test.tar.gz",
}

for more information please see this page, pytorch/text#1756