akanimax / BMSG-GAN

[MSG-GAN] Any body can GAN! Highly stable and robust architecture. Requires little to no hyperparameter tuning. Pytorch Implementation
MIT License
629 stars 105 forks source link

SageMaker config added to GAN training #9

Closed markstrefford closed 5 years ago

markstrefford commented 5 years ago

Conditional use of Sagemaker Reference to PyTorch SageMaker documentation Fixed mod by 0 error for small training sets

markstrefford commented 5 years ago

Fixes as discussed, let me know if you want any more changes (or a clean merge without my history).

akanimax commented 5 years ago

This is perfect :+1:. Thanks a lot.

Cheers :beers:! @akanimax

akanimax commented 5 years ago

resolves #6