Ayush Chaurasia 832ea56eb4 update model initialization design, supports custom data/num_classes (#44) ...

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

2022-11-15 15:36:29 +01:00

..

__init__.py

Segmentation support & other enchancements (#40)

2022-11-08 16:27:57 +01:00

train.py

update model initialization design, supports custom data/num_classes (#44)

2022-11-15 15:36:29 +01:00

val.py

new check_dataset functions (#43)

2022-11-12 15:02:41 +01:00