A Pytorch Implementation of "Attention is All You Need" and "Weighted Transformer Network for Machine Translation"
547
stars
122
forks
source link
only integer tensors of a single element can be converted to an index #4
Open
ShangYuming opened 5 years ago
When I run the colder, there is an error in model.py (line 53):
self.layers = nn.ModuleList( [self.layer_type(d_k, d_v, d_model, d_ff, n_heads, dropout) for _ in range(n_layers)])only integer tensors of a single element can be converted to an index
How can I fix it