search

Appsilon / image_flow_cytometry_fine_tune

MIT License
0 stars 0 forks source link

Fine-tune Swin #7

Open jedrzejwalega opened 1 week ago

jedrzejwalega commented 1 week ago

Now that we've recreated the authors' ResNet18 results (mostly to confirm the lightning and neptune framework works same way as their skorch) we want to try more recent models than ResNet.

The goal is to fine-tune Swin (pretrained can be taken from timm) on our dataset).

We are benchmarking against the authors' val F1 = 0.91.

jedrzejwalega commented 1 week ago

Swin defined here, but not trying it out for now, since we have ConvNext and ViT to work with.