initial oneformer pl datamodule and model loading for trainer

developmentseed / slickformer

Segmenting oil slicks with a panoptic segmentation transformer architecture

4 stars 1 forks source link

Closed rbavery closed 1 year ago

rbavery commented 1 year ago

This PR will set up a train loop for the OneFormer model. So far

pl datamodule set up with Dataloader v1 and datapipes
TODO OneFormer model loading with customized config for 3 class model finetuning
Train loop

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.