mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Pixpro: unwrap losses

Browse Source
This commit is contained in:
James Betker 2021-01-13 11:54:25 -07:00
parent 8990801a3f
commit 038b8654b6
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/pixel_level_contrastive_learning/pixpro_lucidrains.py
View File

@ -489,9 +489,7 @@ class PixelCL(nn.Module):
pix_loss = (loss_pixpro_one_two + loss_pixpro_two_one) / 2
# total loss
loss = pix_loss * self.alpha + instance_loss
return loss, positive_pixel_pairs
return instance_loss, pix_loss, positive_pixel_pairs
# Allows visualizing what the augmentor is up to.
def visual_dbg(self, step, path):