dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
853 Commits 50 Branches 17 Tags 89 MiB
Commit Graph

853 Commits

Author SHA1 Message Date
syl20bnr 2aef3805cc Bump version to 0.12.2 2024-02-01 16:48:47 -05:00
Nathaniel Simard d646417614 Fix docs (#1229) 2024-02-01 15:10:34 -05:00
Nathaniel Simard 23d653ac0b Fix training dashboard metrics switch (#1228) 2024-02-01 15:10:34 -05:00
Sylvain Benner 7a8589e947 Create doc feature in all burn crates and use them for docs.rs build (#1212) 2024-02-01 15:10:34 -05:00
Nathaniel Simard 017f19009d Fix problems with batch norm on LibTorch backend (#1226) 
* Fix problems with batch norm

* Add comment
 2024-02-01 15:10:34 -05:00
Nathaniel Simard 0552b8b73c Fix pow autodiff (#1225) 2024-02-01 15:10:34 -05:00
Nathaniel Simard fb760f49c8 Fix wgpu performance issue: revert to wgpu 0.18.0 (#1221) 
* Use wgpu 0.18.0

* Fix clippy
 2024-02-01 15:10:34 -05:00
Nathaniel Simard 6382aeb7ea Chore: put all dependencies versions in workspace (#1210) 2024-02-01 15:10:33 -05:00
syl20bnr 7b6ac589b0 Bump version to 0.12.1 2024-02-01 15:10:12 -05:00
Louis Fortier-Dubois e03facc5de
fix powf bugs (#1207) 2024-01-31 11:38:45 -05:00
Guillaume Lagrange f1d98bc5f8
Bump tch version to 0.15.0 with libtorch 2.2.0 (#1206) 2024-01-31 11:29:44 -05:00
Aasheesh Singh c620176943
doc fix for mha weights (#1205) 2024-01-31 11:26:38 -05:00
Nathaniel Simard 0ec0fba869
Chore: Update ratatui version (#1204) 2024-01-31 11:26:09 -05:00
dependabot[bot] 22d97e5a59
Bump web-time from 0.2.4 to 1.0.0 (#1199) 
Bumps [web-time](https://github.com/daxpedda/web-time) from 0.2.4 to 1.0.0.
- [Release notes](https://github.com/daxpedda/web-time/releases)
- [Changelog](https://github.com/daxpedda/web-time/blob/main/CHANGELOG.md)
- [Commits](https://github.com/daxpedda/web-time/compare/v0.2.4...v1.0.0)

---
updated-dependencies:
- dependency-name: web-time
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2024-01-31 11:02:35 -05:00
dependabot[bot] d77b189afb
Bump serial_test from 2.0.0 to 3.0.0 (#1203) 
Bumps [serial_test](https://github.com/palfrey/serial_test) from 2.0.0 to 3.0.0.
- [Release notes](https://github.com/palfrey/serial_test/releases)
- [Commits](https://github.com/palfrey/serial_test/compare/v2.0.0...v3.0.0)

---
updated-dependencies:
- dependency-name: serial_test
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2024-01-31 11:02:17 -05:00
Joshua Ferguson 07c69bd29a
Powf fix2 (#1193) 2024-01-31 10:19:15 -05:00
Guillaume Lagrange de84b5c457
Add setup instructions for burn-tch (#1185) 2024-01-31 10:18:09 -05:00
Dilshod Tadjibaev 14aad24c19
Regex fix pytorch (#1196) 2024-01-31 09:53:52 -05:00
Luni-4 c92c51d8b9
ci: Enable dependabot daily (#1195) 2024-01-31 09:48:01 -05:00
Sylvain Benner 02259ea11c
[xtask] New subcommand: Books (#1192) 2024-01-31 09:09:22 -05:00
amfaber ff222b06b5
[burn-train] Fix keys not working reliably to change metrics in terminal UI (#1101) 
* Exit event handler if KeyEvent is of release kind

* Switch logic to return on press

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2024-01-30 22:26:47 -05:00
Nathaniel Simard b7486b0e80
Feat/fusion/wgpu autotune (#1188) 
* wip

* WIP

* Update

* USe read write lock

* Refactor autotune

* Cleanup

* Add docs
 2024-01-30 11:01:43 -05:00
Aasheesh Singh 8b4038d004
Merge pull request #1183 from tracel-ai/docs/guide 
Docs/guide

Improve guide book: Tensor, Dataset and Example Section.
 2024-01-30 10:25:29 -05:00
Aasheesh Singh 4ebab1a538 minor fixes 2024-01-30 10:23:33 -05:00
Luni-4 b8a1c86962
[xtask] Update xtask publish task (#1189) 2024-01-30 10:05:39 -05:00
Roy Varon f5ac5d8e9f
Added tuple modules (#1186) 2024-01-29 17:56:31 -05:00
Dilshod Tadjibaev 67ffa1e54b
Update documentation and book sections on PyTorch import (#1180) 2024-01-29 17:55:20 -05:00
Aasheesh Singh 61dd2c17b9 1. Address comments :) 
2. Fix formatting
3. remove dead link
 2024-01-29 14:39:24 -05:00
Louis Fortier-Dubois 2defd01342
Add tests: Slice assign vs Cat in LSTM backward (#1146) 
* slice assign test

* added tests but no error

* test for non zero grad

* clippy

* i'm confused

* fix ci
 2024-01-28 21:29:06 -05:00
Chen Wu 868608222b
fix: Batchnorm does not work properly when training on multiple devices (#1167) 2024-01-27 10:07:48 -05:00
Joshua Ferguson 4a70a0f8bc
renaming FloatTensor Ops, Primitives, and maybe functions (#1174) 2024-01-27 10:04:50 -05:00
Sylvain Benner 3814c4c9fb
[xtask] New commands: dependencies and vulnerabilities (#1181) 
* [xtask] Add support for vulnerabilities check in xtask

cargo +nightly xtask vulnerability --help

* [xtask] Add support for dependencies check in xtask

cargo xtask dependencies --help

* Make sure all vulnerabilities checks are called with cargo +nightly

* Fix clippy errors

* Use automatic links in docstrings

* Move run function to the top of the file

* Skip dependencies documenation

* pub -> pub(crate)

* Change return type of Sanitizer::flags to &str

* Move the run functions as an impl function of check types enums

* Remove Type suffix in check type enums

* Fix wrong variable name

* cargo_commande -> cargo_crate

* Improve robustness of is_target_supported and add tests

* Reorganize utils module into a directory
 2024-01-27 08:40:41 -05:00
Aasheesh Singh 3e3382c580 fix typo 2024-01-26 06:10:59 -05:00
Aasheesh Singh 9ef9d27012 5. Update dataset.md to explain Dataset transform usages with examples. 2024-01-26 06:05:01 -05:00
Aasheesh Singh 52b11d9a87 4. Add section on Tensor introduction 2024-01-26 04:16:23 -05:00
Aasheesh Singh a5ae89ebee 1. fix broken link in getting-started.md 
2. Make guide documentation and file structure consistent with examples/guide.
3. Add helpful comments in guide.
 2024-01-25 19:17:28 -05:00
Guillaume Lagrange b9bd42959b
Add vision/mnist dataset (#1176) 2024-01-25 16:16:39 -05:00
Dilshod Tadjibaev 0368409eb3
Add support for loading PyTorch `.pt` (weights/states) files directly to model's record (#1085) 2024-01-25 10:20:09 -05:00
unrenormalizable 4ca3e31601
Fixing jupyter nbs + ONNX IR (#1170) 2024-01-24 13:32:01 -05:00
Guillaume Lagrange e1d873abe2
Add Rust concept notes to Burn Book (#1169) 2024-01-24 12:58:14 -05:00
Joshua Ferguson 3b7d9feede
Elementwise pow op (#1133) 2024-01-24 09:46:57 -05:00
Sylvain Benner 707b61051c
Fix doc build on docs.rs (#1168) 2024-01-23 14:13:57 -05:00
Nathaniel Simard eaa4dc3207
Feat/recorder/custom device (#1165) 2024-01-23 13:05:41 -05:00
Guillaume Lagrange e9d1656687
Add model saving and loading examples (#1164) 2024-01-23 12:00:04 -05:00
Nathaniel Simard c9cc1321b6
Update wgpu version (#1166) 2024-01-23 09:50:51 -05:00
Nathaniel Simard 49e49c9bac
Support more elemwise kernel fusion (#1159) 2024-01-22 16:19:20 -05:00
Nathaniel Simard 635cc0a3e7
[burn-fusion] overlapping optimization builders (#1162) 2024-01-22 16:18:59 -05:00
Guillaume Lagrange 191877f14a
New default recorder (#1161) 
* Change DefaultFileRecorder to NamedMpkFileRecorder (no compression)

* Actually, safetensors does not have any checksum for data validation

* Update checksum explainer/recommandation
 2024-01-22 12:50:42 -05:00
Guillaume Lagrange 21772a2662
Fix FullPrecisionSettings type for integers (#1163) 2024-01-22 12:08:56 -05:00
James MacGlashan a752cc172c
Allow raw tensors to be serialized/deserialized directly with serde (#1041) 2024-01-20 11:25:49 -05:00