Using as an image probability model

apple / ml-aim

This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.

Other

1.05k stars 50 forks source link

Using as an image probability model #6

Open alexhepburn opened 10 months ago

alexhepburn commented 10 months ago

I was wondering if you could upload the weights of the final layers used to minimise the NLL, so that this model could be used as an image probability model.