diff --git a/job.py b/job.py index 757d00d67386634c613b628ae68bc7dcd01d42c4..801b842b722539c151d6a06b7e19aba567fbec48 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 8276ecbc60f7153df0e47751cf0af3b36c18f034..b43ba3678e2cada9f1889b1d19c33adc6682a127 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 77dd78228940831bb81b6ed26bb5d18e6060ece7..335fb5bd325cfcef3094b1aa6c663319a679c43a 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