From 738b008da0f3fc919c13c72db145ea7ecfb86a30 Mon Sep 17 00:00:00 2001 From: Javclaude Date: Sat, 30 May 2020 16:41:37 +0200 Subject: [PATCH] Fix comm --- src/model.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/model.py b/src/model.py index 910dea9..a199e1d 100644 --- a/src/model.py +++ b/src/model.py @@ -1,7 +1,6 @@ import torch class LSTMModel(torch.nn.Module): - ''' LSTM model Baseline @@ -16,7 +15,6 @@ class LSTMModel(torch.nn.Module): - n_classes = int, number of classes - do_normalizarion = bool, if true apply layer normalization - pooling_strategy = str, pooling over the hidden dimension - ''' def __init__(self, vocab_size: int,