53 Commits (4198570a4ba98351107527142d85125eb8d0039f)

Author SHA1 Message Date
Ayush Chaurasia 832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
2 years ago
Ayush Chaurasia 1f3aad86c1
new check_dataset functions (#43)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago