iejMac / encoder-distill

Align embedding spaces of PyTorch encoders with common input types.
MIT License
4 stars 0 forks source link

Normalize features before MSE Loss #11

Closed iejMac closed 2 years ago

iejMac commented 2 years ago

I feel like MLP might need to be more complex for this This is why I set bias=True here