martinsbruveris / tensorflow-image-models

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.
https://tfimm.readthedocs.io/en/latest/
Apache License 2.0
287 stars 25 forks source link

Adding convolutional kernel. #95

Closed martinsbruveris closed 1 year ago

martinsbruveris commented 1 year ago

Adding LoRAConv2D layer.

Also, slightly change how we do layer.merge() and layer.unmerge(). Adding and then subtracting the updates does create small numerical errors (you never get exactly back to the starting point). So, instead we now save the previous result and restore that. It does mean, that one shouldn't change the layer while it is in the merged state. But since the merged state is meant only to transfer weights out of the layer, that should be fine.