feat(sagemaker): add pytorch mnist #10354
Conversation
![codeclimate[bot]](https://avatars.githubusercontent.com/in/9403?s=60&v=4){kind=small}
|
|
||
| def get_train_data_loader(batch_size, training_dir, is_distributed, **kwargs): | ||
| logging.info("Get train data loader") | ||
| dataset = datasets.MNIST( |
There was a problem hiding this comment.
Similar blocks of code found in 2 locations. Consider refactoring.
| ) | ||
| test_loader = get_test_data_loader(args.test_batch_size, args.data_dir, **kwargs) | ||
|
|
||
| logging.info( |
There was a problem hiding this comment.
Similar blocks of code found in 2 locations. Consider refactoring.
|
|
||
| def get_test_data_loader(test_batch_size, training_dir, **kwargs): | ||
| logging.info("Get test data loader") | ||
| return torch.utils.data.DataLoader( |
There was a problem hiding this comment.
Similar blocks of code found in 2 locations. Consider refactoring.
| ) | ||
| ) | ||
|
|
||
| logging.info( |
There was a problem hiding this comment.
Similar blocks of code found in 2 locations. Consider refactoring.
| from utils.test import test | ||
|
|
||
|
|
||
| def train(args): |
There was a problem hiding this comment.
Function train has a Cognitive Complexity of 16 (exceeds 5 allowed). Consider refactoring.
|
Code Climate has analyzed commit 9e540f7 and detected 5 issues on this pull request. Here's the issue category breakdown:
View more on Code Climate. |
|
Kudos, SonarCloud Quality Gate passed!
|
![mergify[bot]](https://avatars.githubusercontent.com/in/10562?s=60&v=4){kind=small}
|
Kudos, SonarCloud Quality Gate passed!
|
|
🎉 This PR is included in version 1.85.0 🎉 The release is available on GitHub release Your semantic-release bot 📦🚀 |
No description provided.