search

deep-floyd / IF

Other
7.63k stars 495 forks source link

what is bs_scale for? #145

Open indigopyj opened 8 months ago

indigopyj commented 8 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?