-
Notifications
You must be signed in to change notification settings - Fork 54
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Features/584 data parallel #660
Conversation
…ltz-analytics/heat into features/584-DataParallel
…ltz-analytics/heat into features/584-DataParallel
…ltz-analytics/heat into features/584-DataParallel
…er to reversed computation order
…ltz-analytics/heat into features/584-DataParallel
This branch is updated and the tests run clean. there is no way to increase the coverage for this branch without overhauling how coverage is reported. It is also important to note that since this development, there have been many improvements to the data parallel neural networks which builds on these addition. If there are questions / comments for this PR I will give it until the end of the year, then it will be merged into the master. I apologize for the length of this PR as well. The length was unavoidable for this feature |
…ltz-analytics/heat into features/584-DataParallel Conflicts: heat/nn/__init__.py heat/nn/tests/test_data_parallel.py heat/optim/__init__.py
rerun tests |
1 similar comment
rerun tests |
this is going to be forced in. Other changes are waiting and can be made on the fly |
Description
Implementation of Data Parallel Neural Networks. Documentation about how to use in still being written. This is still not a final iteration, there are many things which will be tuned in the background.
Issue/s resolved: #584 #585 #603 #604 #605 #606
Changes proposed:
DataParallel
classDataLoader
classDataset
class (template)torch.nn
modules andtorch.nn.functional
routinesMNISTDataset
added for loading data and preparing for the network architectureType of change
Due Diligence
All split configurations testedDoes this change modify the behaviour of other functions? If so, which?
no
Notes
There will be backend changes for efficiency coming in the future. however, this feature is becoming rather large and a review/merge would make it more manageable in the future