Skip to content
Snippets Groups Projects
Commit 7a7a1ac6 authored by Maja Jabłońska's avatar Maja Jabłońska Committed by Martyna Wiącek
Browse files

Correct minor mistakes in embeddings.py

parent d79f8224
Branches
Tags
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