Sagemaker environment vars should not be required

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 environment vars should not be required #13

Open powerspowers opened 5 years ago

powerspowers commented 5 years ago

USE_SAGEMAKER is checked at the beginning of train.py but not in the arg code which includes a few sagemaker variables. Easy enough to comment out but would be nice just to pull the USE_SAGEMAKER through completely in the code for all variables.