Added a fine-tuning example for multi-label classification on the Planet dataset sample. Inspired by this post when I was looking for multi-label classification datasets examples.
Refactored resnet-burn to have a virtual workspace and support multiple examples with different dependencies
Moved imagenet.rs module to the inference example since it is not a strict model requirement
Added simple fine-tuning example for ResNet-18 with wgpu and tch-gpu feature flags
I consistently get >92% multi-label accuracy when running the example. Training is pretty quick (even on my peasant laptop GPU w/ 6GB VRAM).
Added a fine-tuning example for multi-label classification on the Planet dataset sample. Inspired by this post when I was looking for multi-label classification datasets examples.
resnet-burn
to have a virtual workspace and support multiple examples with different dependenciesimagenet.rs
module to the inference example since it is not a strict model requirementwgpu
andtch-gpu
feature flagsI consistently get >92% multi-label accuracy when running the example. Training is pretty quick (even on my peasant laptop GPU w/ 6GB VRAM).
Last run: