search

e-lab / ENet-training

351 stars 88 forks source link

Training Enet on a custom dataset #40

Closed souravsingh closed 7 years ago

souravsingh commented 7 years ago

I am looking to train an Enet infrastructure on the dataset available here- https://github.com/jeanpat/DeepFISH

What are the points that I need to take care about when training the dataset on Enet and steps for training the dataset.

codeAC29 commented 7 years ago

@souravsingh you will need to add a data loader. Follow the format used in any of the data loader of this repository and call that data loader here. Just keep in mind that your train/test data and labels follow this format. Also, make sure you have all these return values.