Question about sc_model.py (bs_per_gpu)

[1211sh]

Hi,
I am wondering how the multi-gpu process works in the line 102 and 103.
Line 102 : self.real_A = self.real_A[:bs_per_gpu]

I think self.real_A[:bs_per_gpu] is just simply removing a part of self.real_A, not working as spliting data for multi-gpu.

Is it a typo? or am I wrong?

[o0t1ng0o]

@1211sh
I think that row 102-103 can be deleted?

[lyndonzheng]

Hi @1211sh, this code is used for initializing the feature extractor networks and auxiliary optimizers, instead of the multi-gpu training. The code is based on the CUT.