ignacio-rocco/detectorch

Any plan to update the training file to make it runnable?

Opened this issue · 1 comments

I like this project in the way that it could let me learn pytorch from real implementation. However, The train_fast.py file seems really experimental when I ran it. From the beginning of import mismatches in utils.utils and loss.py files, to the detector initialization (mapping_file, return_rois, return_img_features those parameters are not in the original model definition). I assumed the focus of the project is on evaluations so training is not a priority. But is there any plan in future to update the training file?

+1