CSAILVision / gandissect

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
MIT License
1.77k stars 282 forks source link

How to train own dataset #6

Closed yejg2017 closed 5 years ago

yejg2017 commented 5 years ago

Thanks to your share.I am interested in your job,And I want to know how to train own dataset.Could you help me ?Thanks!

davidbau commented 5 years ago

Check out https://github.com/CSAILVision/semantic-segmentation-pytorch for an example of training a semantic segmentation.