mthorrell / gboost_module

Gradient Boosting Modules for pytorch
MIT License
8 stars 0 forks source link

added input assertion #13

Closed mthorrell closed 5 days ago

mthorrell commented 5 days ago

added input assertion to address current limitations in batch training. See #12 and #9.