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 ...@@ -26,7 +26,7 @@ mkdir $TMPDIR/projet_dl
rsync -r . $TMPDIR/projet_dl rsync -r . $TMPDIR/projet_dl
echo "Checking out the correct version of the code commit_id {commit_id}" 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} git checkout {commit_id}
......
...@@ -59,7 +59,6 @@ class RNN(nn.Module): ...@@ -59,7 +59,6 @@ class RNN(nn.Module):
# One time step # One time step
out, hn = self.rnn(x, h0) out, hn = self.rnn(x, h0)
out = self.fc(out) out = self.fc(out)
print(out.shape)
return out return out
# Bidirectional LSTM # Bidirectional LSTM
......
...@@ -31,7 +31,6 @@ def train(args, model, loader, f_loss, optimizer, device, log_interval = 100): ...@@ -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 # Compute the forward pass through the network up to the loss
outputs = model(inputs) outputs = model(inputs)
print(targets.shape)
loss = f_loss(outputs, targets) loss = f_loss(outputs, targets)
# Backward and optimize # 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