From a929c0c2a2d3b23565cff6f4046dd1af48c327c6 Mon Sep 17 00:00:00 2001 From: Yandi <yandirzm@gmail.com> Date: Tue, 31 Jan 2023 22:32:07 +0100 Subject: [PATCH] RNN_35_16 --- job.py | 2 +- model.py | 1 - train.py | 1 - 3 files changed, 1 insertion(+), 3 deletions(-) diff --git a/job.py b/job.py index 757d00d..801b842 100644 --- a/job.py +++ b/job.py @@ -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} diff --git a/model.py b/model.py index 8276ecb..b43ba36 100644 --- a/model.py +++ b/model.py @@ -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 diff --git a/train.py b/train.py index 77dd782..335fb5b 100644 --- a/train.py +++ b/train.py @@ -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 -- GitLab