search

rosinality / style-based-gan-pytorch

Implementation A Style-Based Generator Architecture for Generative Adversarial Networks in PyTorch
Other
1.1k stars 232 forks source link

How to train it on Custom Data set #97

Open ShakirKhurshid opened 4 years ago

ShakirKhurshid commented 4 years ago

I ve a custom data set of fashion images , will the code run on our custom data set. If yes, where are the modifications required to do that .

rosinality commented 4 years ago

You can use without modifications. You only need to make directory structures that compatible with torchvision.datasets.ImageFolder, that is, you need subdirectory under the path that and image files should be under it.

ShakirKhurshid commented 4 years ago

I actually want to implement on AWS (Sagemaker) and get the data from S3

rosinality commented 4 years ago

Sorry, but I don't know much about that environments.