Skip to content

Commit

Permalink
Merge pull request #660 from helmholtz-analytics/features/584-DataPar…
Browse files Browse the repository at this point in the history
…allel

Features/584 data parallel
  • Loading branch information
coquelin77 authored Feb 26, 2021
2 parents a85de50 + e19138e commit 38ef4a9
Show file tree
Hide file tree
Showing 51 changed files with 2,534 additions and 55 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -304,3 +304,4 @@ tags
MANIFEST
travis_wait_*.log
testing.py
heat/datasets/MNISTDataset
12 changes: 10 additions & 2 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
# Pending additions
- [#717] (https://github.com/helmholtz-analytics/heat/pull/717) Switch CPU CI over to Jenkins and pre-commit to GitHub action.
- [#720] (https://github.com/helmholtz-analytics/heat/pull/720) Ignore test files in codecov report and allow drops in code coverage.
- [#717](https://github.com/helmholtz-analytics/heat/pull/717) Switch CPU CI over to Jenkins and pre-commit to GitHub action.
- [#720](https://github.com/helmholtz-analytics/heat/pull/720) Ignore test files in codecov report and allow drops in code coverage.

## New features
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) NN module for data parallel neural networks
- [#680](https://github.com/helmholtz-analytics/heat/pull/680) New property: larray
- [#683](https://github.com/helmholtz-analytics/heat/pull/683) New properties: nbytes, gnbytes, lnbytes
- [#687](https://github.com/helmholtz-analytics/heat/pull/687) New DNDarray property: balanced
Expand All @@ -20,6 +21,13 @@
- [#679](https://github.com/helmholtz-analytics/heat/pull/679) New feature: ``histc()`` and ``histogram()``
### Linear Algebra
- [#658](https://github.com/helmholtz-analytics/heat/pull/658) Bugfix: `matmul` on GPU will cast away from `int`s to `float`s for the operation and cast back upon its completion. This may result in numerical inaccuracies for very large `int64` DNDarrays
### NN
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) New submodule: `nn.DataParallel` for creating and training data parallel neural networks
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) New feature: Synchronous and Asynchronous gradient updates availble for `ht.nn.DataParallel`
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) New feature: `utils.data.datatools.DataLoader` for created a local `torch.utils.data.Dataloader` for use with `ht.nn.DataParallel`
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) New feature: `utils.data.datatools.Dataset` for created a local `torch.utils.data.Dataset` for use with `ht.nn.DataParallel`
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) Added MNIST example to `example/nn` to show the use of `ht.nn.DataParallel`. The `MNISTDataset` can be found in `ht.utils.data.mnist.py`
- [#660](https://github.com/helmholtz-analytics/heat/pull/660) New feature: Data loader for H5 datasets which shuffles data in the background during training (`utils.data.partial_dataset.PartialH5Dataset`)
### Logical
- [#711](https://github.com/helmholtz-analytics/heat/pull/711) `isfinite()`, `isinf()`, `isnan()`

Expand Down
2 changes: 2 additions & 0 deletions codecov.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@ ignore:
- "heat/core/regression/lasso/demo.py"
- "heat/core/regression/lasso/plotfkt.py"
- "heat/examples/*"
- "heat/utils/data/mnist.py"
- "heat/utils/data/_utils.py"
- "heat/**/test_*.py"

coverage:
Expand Down
Loading

0 comments on commit 38ef4a9

Please sign in to comment.