54 Commits (d3f097314f9478de7f995d4e4b4ccb0c6fbc65d3)

Author SHA1 Message Date
Ayush Chaurasia db1031a1a9
Allow setting model attributes before training (#45)
2 years ago
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