search

tsun / SSRT

Safe Self-Refinement for Transformer-based Domain Adaptation (CVPR 2022)
MIT License
39 stars 7 forks source link

Is that right? #1

Closed cht619 closed 2 years ago

cht619 commented 2 years ago

outputs_tgt = outputs.narrow(0, labels_source.size(0), inputs.size(0)-labels_source.size(0)) # get dt outputs_tgt_perturb = outputs.narrow(0, inputs.size(0), inputs.size(0) - labels_source.size(0)) # what is this for?

cht619 commented 2 years ago

outputs_tgt_perturb = outputs.narrow(0, inputs.size(0), output.size(0) - input.size(0))