#24 F

Merged
jtc merged 1 commits from lsyzz/ccCAN-SAN:test into master 1 year ago
  1. +1
    -1
      models/Backbone.py

+ 1
- 1
models/Backbone.py View File

@@ -239,7 +239,7 @@ class SupConLoss(nn.Module):
loss = - (self.temperature / self.base_temperature) * mean_log_prob_pos
loss = loss.view(anchor_count, batch_size).mean()

loss = 0.05 * loss
loss = 0.1 * loss

return loss



Loading…
Cancel
Save