Skip to content
Snippets Groups Projects
Commit a929c0c2 authored by Yandi's avatar Yandi
Browse files

RNN_35_16

parent 403500f8
No related branches found
No related tags found
1 merge request!1Master into main
......@@ -26,7 +26,7 @@ mkdir $TMPDIR/projet_dl
rsync -r . $TMPDIR/projet_dl
echo "Checking out the correct version of the code commit_id {commit_id}"
cd $TMPDIR/pprojet_dl
cd $TMPDIR/projet_dl
git checkout {commit_id}
......
......@@ -59,7 +59,6 @@ class RNN(nn.Module):
# One time step
out, hn = self.rnn(x, h0)
out = self.fc(out)
print(out.shape)
return out
# Bidirectional LSTM
......
......@@ -31,7 +31,6 @@ def train(args, model, loader, f_loss, optimizer, device, log_interval = 100):
# Compute the forward pass through the network up to the loss
outputs = model(inputs)
print(targets.shape)
loss = f_loss(outputs, targets)
# Backward and optimize
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment