Fix Dependency Name in requirements.txt

[phillipecardenuto]

Thanks for sharing your very interesting work!

I have encountered an installation issue from requirements.txt:

ERROR: No matching distribution found for sklearn==1.1.3

Changing sklearn==1.1.3 to scikit-learn==1.1.3, solved it.
So, I am suggesting updating requirements.txt to prevent this error for future installations.

Thanks again for the great work!
Kind Regards,
Joao

[phillipecardenuto]

In addition to the listed dependencies, I found it necessary to include both tensorflow and timm in my setup, opting for the latest tensorflow version and timm==0.6.12

However, even with that, I couldn't load the model properly.

RuntimeError: Error(s) in loading state_dict for DataParallel:
Unexpected key(s) in state_dict: "module.head.2.weight", "module.head.2.bias".
size mismatch for module.head.0.weight: copying a param with shape torch.Size([2048, 2048]) from checkpoint, the shape in current model is torch.Size([128, 2048]).
size mismatch for module.head.0.bias: copying a param with shape torch.Size([2048]) from checkpoint, the shape in current model is torch.Size([128]).

[avinabsaha]

Try using --head mlp!

[phillipecardenuto]

Thanks, I missed that.
It's working now!