Where is the UNI model code?

[billy-enrizky]

Dear Faisal Mahmood Lab staffs,

I have seen the code and from what I understand, you built the UNI model based on ResNet50. Then, you trained the model based on millions of images.

Is it correct that you did not establish new model architecture?

[guillaumejaume]

Hi @billy-enrizky,

UNI is based on a Vision Transformer Large model, not ResNet50. The model can be loaded using Trident model factory, see https://github.com/mahmoodlab/TRIDENT/blob/main/trident/patch_encoder_models/load.py#L15.