pixel-constrained-cnn-pytorch icon indicating copy to clipboard operation
pixel-constrained-cnn-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon Schlumberger

Training network on own data

Open yh675 opened this issue 4 years ago • 1 comments

Hi,

Thanks for sharing this cool work!

Is it possible to train the network on my own dataset?

yh675 avatar Sep 18 '20 16:09 yh675

Hi, thank you! Yes it should definitely be possible to train the model on your own dataset. You can add a dataloader for your own dataset (in utils/dataloaders.py) and then import it in the main.py script. The model should then train on your dataset.

EmilienDupont avatar Sep 21 '20 09:09 EmilienDupont