Open hdmjdp opened 1 year ago
https://github.com/mindslab-ai/phaseaug/blob/07d1c74244fc55359d022bfb1bd21acce2cc9066/train.py#LL175C8-L175C20
I think aug is frozened, not trained.
It does not contain any trainable variable, so identical to frozen. So it is identical results even if train() or eval().
train()
eval()
I added it for clarity.
https://github.com/mindslab-ai/phaseaug/blob/07d1c74244fc55359d022bfb1bd21acce2cc9066/train.py#LL175C8-L175C20
I think aug is frozened, not trained.