GitHub - lucidrains/tab-transformer-pytorch: Implementation ...
github.com › lucidrains › tab-transformer-pytorchDec 15, 2020 · import torch import torch. nn as nn from tab_transformer_pytorch import TabTransformer cont_mean_std = torch. randn (10, 2) model = TabTransformer ( categories = (10, 5, 6, 5, 8), # tuple containing the number of unique values within each category num_continuous = 10, # number of continuous values dim = 32, # dimension, paper set at 32 dim_out ...
GitHub - tunz/transformer-pytorch: Transformer ...
https://github.com/tunz/transformer-pytorch07.03.2019 · Transformer. This is a pytorch implementation of the Transformer model like tensorflow/tensor2tensor. Prerequisite. I tested it with PyTorch 1.0.0 and Python 3.6.8. It's using SpaCy to tokenize languages for wmt32k dataset. So, if you want to run wmt32k problem which is a de/en translation dataset, you should download language models first with the following …