Skip to content
Snippets Groups Projects
Commit 4a3f4da8 authored by Maja Jabłońska's avatar Maja Jabłońska
Browse files

Correct minor mistakes in embeddings.py

parent 7496eabb
1 merge request!46Merge COMBO 3.0 into master
from typing import Optional
from typing import Optional, List
import torch
from overrides import overrides
......@@ -151,7 +151,7 @@ class _TorchEmbedder(TokenEmbedder):
embedded = self._embedding(x)
if self._projection:
projection = self._projection
for _ in range(embedded.dim()-2):
for p in range(embedded.dim()-2):
projection = TimeDistributed(p)
embedded = projection(embedded)
return embedded
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment