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

165 Commits

Author SHA1 Message Date
Jakub 3592f3799a
fix(book): add missing second parameter to CrosEntropyLoss constructor (#1301) 
* fix(book): add missing second parameter to CrosEntropyLoss constructor

CrossEntropyLoss::new() expects two parameters, the pad_index and the device

* fix: fix missing closing parenthese
 2024-02-15 09:46:41 -05:00
Jakub a68b494531
fix(book): add missing device parameter to mode.init() (#1302) 2024-02-13 09:34:03 -05:00
Nathaniel Simard a9b6dbc177
Enable burn fusion by default (#1223) 2024-02-08 11:28:02 -05:00
Guillaume Lagrange 059c241511
Add vision feature for MNIST in book guide (#1267) 2024-02-07 09:21:35 -05:00
Sylvain Benner 4aa13d6b25
Bump Burn version to 0.13 (#1211) 2024-01-31 16:01:20 -05:00
Dilshod Tadjibaev 14aad24c19
Regex fix pytorch (#1196) 2024-01-31 09:53:52 -05:00
Sylvain Benner 02259ea11c
[xtask] New subcommand: Books (#1192) 2024-01-31 09:09:22 -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
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
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
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
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
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 6079f98950
Fix missing ticks in Burn book and remove unused example dependency (#1144) 
* Add missing ticks for triu and tril ops

* Removed example unused dependency
 2024-01-16 09:03:17 -05:00
Guillaume Lagrange 15c2f3b7a1
Fix small typos and dead links in README and burn-book (#1127) 2024-01-10 08:57:28 -05:00
Kirill Mavreshko 97297538b1
Remove _devauto fuctions (#518) (#1110) 2024-01-06 13:36:34 -05:00
Nathaniel Simard fceb036c6f
Fix: serde dependency (#1091) 
* Re-export serde

* Fix

* USe another strategy

* Fix

* Fix

* Update de book
 2023-12-22 16:53:34 -05:00
Kirill Mavreshko 1fd07fcb4a
Explicit device tensors (#1081) 2023-12-20 17:49:59 -05:00
pacowong c1cb77ac2e
Add mish and softplus activation function (#1071) 2023-12-20 12:41:23 -05:00
Louis Fortier-Dubois 8fc52113bc
Chore/bump v12 (#1048) 2023-12-04 10:47:54 -05:00
Louis Fortier-Dubois 3088c466a5
patch 0.11.1 (#1047) 2023-12-04 10:18:30 -05:00
Timothy 52811f9938
Fix some issues in Burn book (#1042) 2023-12-04 09:23:47 -05:00
Nathaniel Simard ab1b5890f5
Chore/release (#1031) 2023-12-01 14:33:28 -05:00
Maksym Hazevych 87393b2070
Updated explanation to match code (#1012) 
Updated the explanation at the beginning to match the actual changes to 'Cargo.toml'. Moved down the 'edition' property to match the style of the initial 'Cargo.toml' file. Removed unnecessary comment for a self-explanatory piece of code. Some rewording.
 2023-11-29 09:41:07 -05:00
David Chavez fda9e5d388
feature(tensor): Add triu and tril ops (#1002) 2023-11-29 09:39:42 -05:00
David Chavez 929b1786bb
feature(tensor): Add chunk op (#998) 2023-11-27 09:58:43 -05:00
David Chavez 2fdf9a3591
feature(tensor): Add Tensor::stack (#1004) 2023-11-27 09:57:31 -05:00
Granville Schmidt d0cb7205fa
fix typo in basic-workflow training (#1000) 2023-11-27 09:37:29 -05:00
David Chavez f09baadfc5
feature(tensor): Add narrow op (#996) 
* Add narrow methods

* Revert "Add narrow methods"

This reverts commit 9371d87c79.

* Implement a shared version of narrow

* Correct test case

* Update book

* Improve tests
 2023-11-24 08:24:06 -05:00
Dilshod Tadjibaev be5bb33788
Update getting-started.md (#987) 
Fixes #828
 2023-11-21 17:40:40 -05:00
Sylvain Benner 2f079e991b
Fix burn version in the Burn book snippet (#985) 
The book assumes the latest version which is 0.10.0 at the time of this commit.
 2023-11-21 15:59:25 -05:00
David Chavez 88b442015d
feature(norm): Add GroupNorm (#963) 
* Add GroupNorm

* Fix implemenation and add tests

* Address PR comments

* Fix formatting

* Update burn book
 2023-11-21 09:22:56 -05:00
David Chavez 49e16b6834
feature(tensor): Add unsqueeze_dim helper (#966) 2023-11-20 10:29:40 -05:00
Rodrigo Pandini 20e9066b57
fix with_file_checkpointer use in training example (#973) 2023-11-20 09:17:29 -05:00
Zsombor 4fc0c27e31
Implement tensor.recip() function to calculate elementwise reciprocals (#953) 2023-11-15 09:17:32 -05:00
dengelt e255472826
Fix typos (#949) 2023-11-12 15:08:39 -05:00
Nathaniel Simard 96524d40a1
[Breaking] Refactor Backend Names (#904) 2023-10-29 18:27:49 -04:00
Nathaniel Simard e581912c8a
Update book (#858) 2023-10-11 09:11:28 -04:00
Nathaniel Simard 620b86de98
Feat training events (#857) 2023-10-10 13:27:03 -04:00
Peter Szilvasi 6c32e23f15
Fix the device initialization in backend.md (#808) 2023-09-16 09:44:34 -04:00
Louis Fortier-Dubois d249d24c07
simplify getting started (#782) 2023-09-07 14:57:32 -04:00
Dilshod Tadjibaev 27952b1f47
Fix broken link and other minor fixes (#780) 2023-09-07 08:44:57 -04:00
Nathaniel Simard 8b3d10c4d3
Cleanup of the book + feature flags (#773) 
---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-09-06 09:16:36 -04:00
Nathaniel Simard ab4d2f8d63
Book: Add Config, Dataset and Record building blocks (#770) 2023-09-05 17:25:05 -04:00
Louis Fortier-Dubois 07e8ee948d
update readme for next release (#769) 2023-09-05 16:42:22 -04:00
Louis Fortier-Dubois 56b8e29514
Doc/building blocks more (#745) 2023-09-02 13:09:03 -04:00
Dilshod Tadjibaev 0f0e66a162
Add import onnx model section to the book (#750) 2023-09-02 11:24:49 -04:00
Louis Fortier-Dubois acfd1e0fe9
Doc/book/getting started (#744) 2023-08-31 14:08:13 -04:00
Dilshod Tadjibaev 510d2d4cb3
Apply markdown formatting using prettier (#743) 2023-08-31 11:44:29 -04:00
Nathaniel Simard c95b34c511
Book: backend extension + custom wgpu kernel (#728) 2023-08-31 09:55:43 -04:00
Dilshod Tadjibaev 40a6573f28
Book: Expanded the planned sections and added built-in module section (#730) 2023-08-30 12:39:35 -04:00
Asuka Minato 0f7864f11b
Path type (#699) 2023-08-27 09:17:48 -04:00
Mahmoud b263faebc3
fix(book): use the correct burn versions (#681) 2023-08-23 14:07:25 -04:00
Nathaniel Simard 00d3d208b8
Add book + fix some code (#671) 2023-08-23 11:52:55 -04:00
Dilshod Tadjibaev 8448611908
License fixes (#648) 2023-08-16 12:45:35 -04:00
Louis Fortier-Dubois aebd359e66
Book/guide (#632) 2023-08-14 11:58:27 -04:00
Wouter Doppenberg f819898097
The Burn Book (#625) 2023-08-10 14:17:23 -04:00