Train pytorch model for UK moths

AMI-system / species_classifier

This repository contains the code to create on-device machine learning models for species classification.

MIT License

2 stars 0 forks source link

Closed KatrionaGoldmann closed 1 year ago

KatrionaGoldmann commented 1 year ago

Sub-tasks:

Issues arising:

KatrionaGoldmann commented 1 year ago

Use list of moths provided in #4

KatrionaGoldmann commented 1 year ago

[x] classifying three genus (abraxas, Acasis, Achlya), five species
[x] all species currently crashing for all species on M1 mac, I suspect this needs to be run on a more powerful server/VM to run on basekerville with SLURM

KatrionaGoldmann commented 1 year ago

working for https://github.com/AMI-trap/on_device_classifier/commit/eca6238d301f45580e8d1ea56789c36d256a189f (on baskerville see on_device_classifier/02_model_training/pytorch/logs/turing-macro_v01_efficientnetv2-b3_2023-06-08-14-34.pt)