CCInc / 3d-ml

A versatile framework for 3D machine learning built on Pytorch Lightning and Hydra [looking for contributors!]
15 stars 3 forks source link

Refactor Generic Model Code by Task #4

Closed CCInc closed 2 years ago

CCInc commented 2 years ago

Have generic base models for tasks such as classification and segmentation, which commonly share the same metrics (such as loss, accuracy, and iou). This would likely entail refactoring out most methods besides the step and forward logic, to preprocess the data for the downstream model plugin as needed.