deep-floyd / IF

Other
7.69k stars 501 forks source link

what is bs_scale for? #145

Open indigopyj opened 11 months ago

indigopyj commented 11 months ago

https://github.com/deep-floyd/IF/blob/ffc8163891682beaca0c5eb6b9077860a3bc6509/deepfloyd_if/modules/base.py#L99

I am interested in inpainting model and I found the variable 'bs_scale'. However, I don't know what is the purpose of this variable. Why do we have to multiple batch_size by 2 ?

Does anyone know about it?