Update train.py

2020-03-02 07:33:25 -03:00 · 2020-03-02 07:33:25 -03:00 · cdb4a0dd97
parent 542b915e6e
commit cdb4a0dd97
1 changed files with 0 additions and 5 deletions
--- a/train.py
+++ b/train.py
@ -55,11 +55,6 @@ n_iter = 0
 for e in range(config['epochs']):
    for step, ((xis, xjs), _) in enumerate(train_loader):

-        # fig, axs = plt.subplots(nrows=1, ncols=2, constrained_layout=False)
-        # axs[0].imshow(xis[2].numpy().transpose((1,2,0)))
-        # axs[1].imshow(xjs[2].numpy().transpose((1,2,0)))
-        # plt.show()
-
        optimizer.zero_grad()

        if train_gpu: