search

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
286 stars 25 forks source link

Adding classification layer to training #92

Closed kevin-keraudren closed 1 year ago

kevin-keraudren commented 1 year ago

We need to include the weights of the final classification layer in the training variables.