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

735 Commits

Author SHA1 Message Date
Louis Fortier-Dubois fb2a71bb81
remove to device (#694) 2023-08-25 09:55:18 -04:00
Will Brickner bab036b263
Added custom initializer option for MHA and other transformer-related modules (#693) 2023-08-25 09:55:05 -04:00
Louis Fortier-Dubois 88cb6b07fc
Feat/candle/more operations (#682) 2023-08-25 08:46:30 -04:00
Nathaniel Simard 481ff14fe1
feat: can add custom training and validation metric loggers (#690) 2023-08-25 07:16:45 -04:00
Nathaniel Simard efee0ac296
Feat/train/custom optimize method (#689) 
* Add the possibility to add a custom optimize function for models

* Fix clippy
 2023-08-25 07:14:36 -04:00
Nathaniel Simard 183620fb20
feat: replace grad tensor (#688) 2023-08-24 14:07:04 -04:00
Jerome Robert edb3e9fc4b
Do not use default device when running kernel::matmul::tune (#684) 
Use the device of the involved Tensor instead of Device::default
 2023-08-24 14:01:27 -04:00
Louis Fortier-Dubois bc27a87e9d
new clippy stuff (#687) 2023-08-24 13:20:58 -04:00
Mahmoud b263faebc3
fix(book): use the correct burn versions (#681) 2023-08-23 14:07:25 -04:00
Nathaniel Simard d18d1b0bb9
Can configure wgpu max tasks (#603) 2023-08-23 12:20:27 -04:00
Nathaniel Simard 00d3d208b8
Add book + fix some code (#671) 2023-08-23 11:52:55 -04:00
Mahmoud b60d931771
chore: update README.md (#678) 2023-08-23 10:32:51 -04:00
dependabot[bot] 8f4faabd5f
Update gix-tempfile requirement from 7.0.0 to 8.0.0 (#679) 
Updates the requirements on [gix-tempfile](https://github.com/Byron/gitoxide) to permit the latest version.
- [Release notes](https://github.com/Byron/gitoxide/releases)
- [Changelog](https://github.com/Byron/gitoxide/blob/main/CHANGELOG.md)
- [Commits](https://github.com/Byron/gitoxide/commits)

---
updated-dependencies:
- dependency-name: gix-tempfile
  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-23 10:32:30 -04:00
Will Brickner d1a708e317
Made TchTensor fields public (#677) 2023-08-22 21:37:58 -04:00
Caio Piccirillo 2fefc82099
Dilation maxpool (#668) 2023-08-21 14:14:25 -04:00
Elazrod56 dd5ea5251c
Training metrics (#647) 2023-08-21 14:13:36 -04:00
Louis Fortier-Dubois b07af74788
support broadcast matmul (#669) 2023-08-21 11:43:21 -04:00
Dilshod Tadjibaev b06fe660e8
Onnx tests and bug fixes - BatchNorm, Identity, Relu, Sigmoid and Transpose (#661) 2023-08-21 11:39:59 -04:00
Baye Dieng a557caa4f7
remove channel field from maxpool1d config struct (#666) 2023-08-20 19:24:24 -04:00
Louis Fortier-Dubois 6a5ea0ef7c
Feat/candle/basic operations (#664) 2023-08-20 18:55:14 -04:00
nathaniel 6b5ba77084 Fix build 2023-08-17 11:13:59 -04:00
Nathaniel Simard bda03c6a76
Feat/avg pool/include pad config (#653) 2023-08-17 08:50:31 -04:00
Louis Fortier-Dubois c1eddf04fc
Feat/candle/initialize (#650) 2023-08-17 08:50:08 -04:00
Dilshod Tadjibaev feabc14d2e
Fix Conv1d kernel for ONNX importing (#658) 2023-08-17 08:41:22 -04:00
AuruTus aca31d0b31
fix: common weight_decay method (#656) 2023-08-17 08:38:57 -04:00
Dilshod Tadjibaev 8808ee265e
Add typos to checks script (#649) 2023-08-16 18:44:16 -04:00
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