mirrors
/
SimCLR
mirror of https://github.com/sthalles/SimCLR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update train.py

pull/5/head
Thalles Silva 2020-03-02 07:33:25 -03:00 committed by GitHub
parent 542b915e6e
commit cdb4a0dd97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -55,11 +55,6 @@ n_iter = 0
for e in range(config['epochs']): for e in range(config['epochs']):
for step, ((xis, xjs), _) in enumerate(train_loader): 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() optimizer.zero_grad()
if train_gpu: if train_gpu: