Question about the memory bank and the keys

Haochen-Wang409 / U2PL

[CVPR'22] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels

Apache License 2.0

426 stars 59 forks source link

Question about the memory bank and the keys #133

Closed Hugo-cell111 closed 1 year ago

Hugo-cell111 commented 1 year ago

Hi! In this line of code, the memobank only serves as a parameter of the function "compute_contra_memobank_loss" and it hasn't been renewed during the training process. Plus, the parameter "new_keys" is just returned without using it. So is there any mistake here? Thanks!

Haochen-Wang409 commented 1 year ago

The memory bank is updated using enqueue_and_dequeue.