53 Commits (1cb92d7f42a3725a0114dd60df12c059c495c5eb)

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