dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
759 Commits 39 Branches 19 Tags 101 MiB
Commit Graph

759 Commits

Author SHA1 Message Date
Dilshod Tadjibaev 0a1ba971cc
Fix bug related to calculation of flatten op output dim (#652) 
Fixes #646
 2023-08-16 18:22:23 -04:00
Dilshod Tadjibaev 294a319557
Add Jupyter notebook examples (#651) 2023-08-16 18:17:12 -04:00
Dilshod Tadjibaev 8448611908
License fixes (#648) 2023-08-16 12:45:35 -04:00
AuruTus 3e4adc4bc7
Feat/RMSProp-optimizer (#607) 2023-08-16 11:43:47 -04:00
Dilshod Tadjibaev 3264b1007c
Changes and fixes required for squeezenet1.1-opset7 onnx (#639) 2023-08-15 15:37:52 -04:00
Louis Fortier-Dubois d659f11639
Perf/wgpu/autotune (#609) 2023-08-15 11:26:00 -04:00
dependabot[bot] c6c7a4cc48
Bump crate-ci/typos from 1.16.4 to 1.16.5 (#643) 
Bumps [crate-ci/typos](https://github.com/crate-ci/typos) from 1.16.4 to 1.16.5.
- [Release notes](https://github.com/crate-ci/typos/releases)
- [Changelog](https://github.com/crate-ci/typos/blob/master/CHANGELOG.md)
- [Commits](cad89d4596...0863e64406)

---
updated-dependencies:
- dependency-name: crate-ci/typos
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-15 09:28:50 -04:00
Mitchell Robert Vollger 434975c410
Add conv1d to burn-import (onnx) (#638) 2023-08-14 17:48:02 -04:00
Luni-4 7e4feb7d8b
Convert `run-checks` script into a Rust binary (#628) 2023-08-14 17:43:21 -04:00
Ariel Cohen d8ea723ea0
Add adagrad (#627) 2023-08-14 17:14:33 -04:00
Luni-4 fefc63895d
Improve `publish` binary (#642) 2023-08-14 16:47:57 -04:00
nathaniel 8430d187ba Update tokenizers version 2023-08-14 14:01:16 -05:00
nathaniel 8e55f60676 Fix in text classification 2023-08-14 14:01:16 -05:00
nathaniel 2203eebd70 Fix CI 2023-08-14 14:01:16 -05:00
dependabot[bot] 343f7f5be7
Bump crate-ci/typos from 1.16.3 to 1.16.4 (#640) 
Bumps [crate-ci/typos](https://github.com/crate-ci/typos) from 1.16.3 to 1.16.4.
- [Release notes](https://github.com/crate-ci/typos/releases)
- [Changelog](https://github.com/crate-ci/typos/blob/master/CHANGELOG.md)
- [Commits](ef5fcf92df...cad89d4596)

---
updated-dependencies:
- dependency-name: crate-ci/typos
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-14 12:21:56 -04:00
Louis Fortier-Dubois aebd359e66
Book/guide (#632) 2023-08-14 11:58:27 -04:00
Louis Fortier-Dubois 8770368892
bugfix: Revert last Adam bugfix (#630) 2023-08-14 09:27:24 -04:00
Luni-4 b47e7bea18
Fix `publish` binary (#629) 2023-08-14 09:26:18 -04:00
Wouter Doppenberg f819898097
The Burn Book (#625) 2023-08-10 14:17:23 -04:00
nathaniel 2709e4c36e Add new sponsor 2023-08-10 10:19:53 -04:00
dependabot[bot] fc57c266fc
Bump crate-ci/typos from 1.16.2 to 1.16.3 (#624) 
Bumps [crate-ci/typos](https://github.com/crate-ci/typos) from 1.16.2 to 1.16.3.
- [Release notes](https://github.com/crate-ci/typos/releases)
- [Changelog](https://github.com/crate-ci/typos/blob/master/CHANGELOG.md)
- [Commits](8a7996b4bc...ef5fcf92df)

---
updated-dependencies:
- dependency-name: crate-ci/typos
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-10 09:35:35 -04:00
Dilshod Tadjibaev b79fa0748f
Fix bugs with MaxPool2d in ONNX conversation (#623) 2023-08-10 09:35:15 -04:00
Wouter Doppenberg 4c663b4cb7
Added correct expected weights, fixed adam state init (#621) 2023-08-09 17:55:39 -04:00
Dilshod Tadjibaev c1ba355c26
Softmax OP ONNX support (#620) 2023-08-09 16:48:45 -04:00
Nathaniel Simard c74e75f748
Fix/wgpu/max pool2d backward (#613) 2023-08-09 16:45:49 -04:00
Dilshod Tadjibaev 894783f08d
Global avg pool (#611) 2023-08-09 16:15:33 -04:00
Caio Piccirillo cb283a9e5b
Max pool1d (#602) 2023-08-09 16:13:48 -04:00
Wouter Doppenberg 1f01fcb640
AdamW implementation (#566) 2023-08-09 16:09:17 -04:00
Dilshod Tadjibaev 716b7569de
Add Dropout ONNX OP and other refactoring (#606) 2023-08-09 09:53:20 -04:00
Luni-4 2259b6222e
scripts: Remove an unused script (#610) 
Since 0.8.0 version has been released, this script is no more necessary
 2023-08-09 09:35:27 -04:00
Caio Piccirillo 1d3bbaab13
Typos (#608) 2023-08-08 17:57:51 -04:00
Nathaniel Simard 441a7011ce
Feat/tensor casting (#604) 2023-08-08 10:02:17 -04:00
Nathaniel Simard 8bc687e1bb
WGPU use best limits for the adaptor (#601) 2023-08-07 15:10:21 -04:00
Ikko Eltociear Ashimine 9361193b5d
Fix typo in tensor.rs (#599) 
overriden -> overridden
 2023-08-07 12:28:08 -04:00
Gadersd ed255c5561
Use buffered io for massive performance gains when loading and saving… (#593) 2023-08-06 12:56:27 -04:00
Nathaniel Simard 3d9f1894ae
Link models repository (#596) 2023-08-06 12:53:33 -04:00
mashirooooo 72f6bf3ca7
feat: check python venv environment initialized (#588) 2023-08-06 11:49:35 -04:00
Dilshod Tadjibaev 5cc32cc8cb
Remove topological sort of nodes (#589) 
ONNX nodes should come as topologically sorted, so we are removing it instead, we are making sure the nodes are topologically sorted.
 2023-08-06 10:51:28 -04:00
Nathaniel Simard ce8a175aa4
Feat/conv transpose1d backward (#586) 2023-08-06 10:50:10 -04:00
Dilshod Tadjibaev 1554a3c898
Full support for ONNX scalar operators and Constants (#578) 2023-08-04 16:51:51 -04:00
Nathaniel Simard ca9a8808d9
Feat/adaptive avg pool1d (#585) 2023-08-04 13:55:18 -04:00
Nathaniel Simard 8436d4ff66
Feat/tensor/adaptive avg pool2d (#572) 2023-08-04 10:23:59 -04:00
Nathaniel Simard 597eab524d
Feat/conv transpose2d (#574) 2023-08-03 15:42:18 -04:00
Chris Cowan 706a6ea99e
Fix huggingface downloader Windows support (#564) (#575) 2023-08-03 09:23:12 -04:00
dependabot[bot] b7ad23bd87
Update serial_test requirement from 0.5.0 to 2.0.0 (#579) 
Updates the requirements on [serial_test](https://github.com/palfrey/serial_test) to permit the latest version.
- [Release notes](https://github.com/palfrey/serial_test/releases)
- [Commits](https://github.com/palfrey/serial_test/compare/v0.5.0...v2.0.0)

---
updated-dependencies:
- dependency-name: serial_test
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-03 09:21:47 -04:00
Louis Fortier-Dubois d5f9f69cea
Refactor/wgpu/prng (#576) 2023-08-02 16:08:50 -04:00
mmalczak 73fb0eaa7e
Addition of abs tensor opperator #506 (#553) 2023-08-01 18:25:14 -04:00
Louis Fortier-Dubois 87125da6c9
Feat/wgpu/prng bernoulli (#571) 2023-08-01 12:54:22 -04:00
Louis Fortier-Dubois a69788ad4b
Feat/wgpu/prng normal (#570) 2023-08-01 10:59:49 -04:00
Louis Fortier-Dubois daedec6f6d
Feat/wgpu/prng (#560) 
* wip

* wip

* default prng working but not perfectly random

* format

* format

* fix prng algo

* clippy

* uniform prng

* refactor tests for prng

* cleanup
 2023-08-01 09:26:49 -04:00