Skip to content

Commit

Permalink
Feature/manifold mixup (#644)
Browse files Browse the repository at this point in the history
* Add manifold mixup

* Update and cite

* Fix bug

* Update

* Update readme
  • Loading branch information
MattPainter01 authored and ethanwharris committed Nov 6, 2019
1 parent 3335931 commit 00af4ff
Show file tree
Hide file tree
Showing 6 changed files with 964 additions and 2 deletions.
3 changes: 2 additions & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
### Added
- Added BCPlus callback for between-class learning
- Added support for PyTorch 1.3
- Added a show flag to the `ImagingCallback.to_pyplot` method, set to false to stop it from calling `plt.show`
- Added a show flag to the `ImagingCallback.to_pyplot` method, set to false to stop it from calling `plt.show`
- Added manifold mixup
### Changed
- Changed the default behaviour of `ImagingCallback.to_pyplot` to turn off the axis
### Deprecated
Expand Down
26 changes: 25 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -407,7 +407,7 @@ A PyTorch model fitting library designed for use by researchers (or anyone reall
</a>
</td>
</tr>
<tr>
<tr>
<td align="center">
<a href="https://github.com/pytorchbearer/torchbearer/blob/master/docs/_static/notebooks/regularisers.ipynb">
<img src="http://www.pytorchbearer.org/assets/img/github_logo.png" height="32">
Expand All @@ -421,6 +421,30 @@ A PyTorch model fitting library designed for use by researchers (or anyone reall
</a>
</td>
</tr>
<tr>
<td rowspan="3" colspan="2">
<b>Manifold Mixup:</b> A demonstration of how to use the Manifold Mixup callback in Torchbearer.
</td>
<td align="center">
<a href="https://nbviewer.jupyter.org/github/pytorchbearer/torchbearer/blob/master/docs/_static/notebooks/manifold_mixup.ipynb">
<img src="http://www.pytorchbearer.org/assets/img/nbviewer_logo.svg" height="34">
</a>
</td>
</tr>
<tr>
<td align="center">
<a href="https://github.com/pytorchbearer/torchbearer/blob/master/docs/_static/notebooks/manifold_mixup.ipynb">
<img src="http://www.pytorchbearer.org/assets/img/github_logo.png" height="32">
</a>
</td>
</tr>
<tr>
<td align="center">
<a href="https://colab.research.google.com/github/pytorchbearer/torchbearer/blob/master/docs/_static/notebooks/manifold_mixup.ipynb">
<img src="http://www.pytorchbearer.org/assets/img/colab_logo.png" height="28">
</a>
</td>
</tr>
<tr>
<td rowspan="3">
<img src="http://www.pytorchbearer.org/assets/img/examples/cam.jpg" width="256">
Expand Down
Loading

0 comments on commit 00af4ff

Please sign in to comment.