Commit Graph

759 Commits

Author SHA1 Message Date
Dilshod Tadjibaev bdcd5873c5
Support no-std for models generated by burn-import (#822) 2023-09-21 11:43:07 -04:00
Dilshod Tadjibaev 35ca839e18
Add support for different record types in ONNX (#816) 2023-09-21 09:06:57 -04:00
Juliano Decico Negri 293020aae6
#384 Include tests for int.rs and float.rs (#794) 2023-09-21 09:00:09 -04:00
Dilshod Tadjibaev 393d86e99d
Fix dataset count slowness (#818)
Fixes #811
2023-09-21 08:55:13 -04:00
Nathaniel Simard aacf191161
Fix training checkpoints (#815) 2023-09-21 08:52:04 -04:00
Nathaniel Simard ac4adb54ea
Burn compute (#809) 2023-09-18 19:56:53 -04:00
Damien Elmes d7e9e75099
Fix train-minimal feature and ensure it gets tested (#802) 2023-09-16 09:52:14 -04:00
Peter Szilvasi 6c32e23f15
Fix the device initialization in backend.md (#808) 2023-09-16 09:44:34 -04:00
Luni-4 5691faa3c2
ci: Improve cache key to better discriminate among caches (#804) 2023-09-15 09:42:49 -04:00
Arvid Hammarlund 8f65acf65a
Made docs show features not included in std. (#789) 2023-09-15 09:14:44 -04:00
Nathaniel Simard 57d6a566be
Feat/dashboard tui (#790) 2023-09-13 10:45:14 -04:00
Nathaniel Simard 4f72578260
Fix compilation flags (#800) 2023-09-13 10:38:16 -04:00
Damien Elmes 6187d88226
Check examples when all tests are run (#801)
Will break CI until #800 is merged.
2023-09-13 09:56:56 -04:00
Nathaniel Simard 251ec00070
Fix clamp API (#799) 2023-09-12 14:20:11 -04:00
Damien Elmes dbc5f0a340
Fix missing train re-export when using train-minimal (#785)
Accidental regression in #773
2023-09-08 09:04:09 -04:00
Justin Moore d2278002e4
Avoid building `xtask` when invoking `cargo build` (#784) 2023-09-08 09:01:39 -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
Luni-4 c0eb339a8e
Remove binaries from .gitignore (#775) 2023-09-07 08:43:52 -04:00
Nathaniel Simard af0be5cfeb
Chore: bump version (#777) 2023-09-06 12:15:13 -04:00
nathaniel 76894eff04 Remove candle from release 2023-09-06 10:25:37 -04:00
Nathaniel Simard 8be5889367
Prepare next release (#774)
* Prepare for next release

* Fix
2023-09-06 09:29:05 -04:00
Arvid Hammarlund f330f9f73d
Added Dim-iter (#771)
* "added dim-iter".

* added tests to iter_dim

* Made the rnn implemenations use squeeze.
2023-09-06 09:22:30 -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
Justin Moore 6095dd104e
Add `cargo-xtask` helper and move scripts into it (#757) 2023-09-06 08:22:00 -04:00
Nathaniel Simard ab4d2f8d63
Book: Add Config, Dataset and Record building blocks (#770) 2023-09-05 17:25:05 -04:00
Arvid Hammarlund a20a1a8cee
Fixed bug in binary cross entropy (#772) 2023-09-05 17:07:07 -04:00
Louis Fortier-Dubois 07e8ee948d
update readme for next release (#769) 2023-09-05 16:42:22 -04:00
Luni-4 359dc2733e
Remove run-checks binary (#768) 2023-09-05 10:20:09 -04:00
Nathaniel Simard 3aed8571d4
Fix: allow module with custom backend trait (#761) 2023-09-05 10:19:19 -04:00
Nathaniel Simard a9596af07a
Doc: improve config builder docs (#760)
* Doc: improve config builder docs

* fix ci

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
2023-09-05 10:16:55 -04:00
Damien Elmes 2b96c73898
MSRV bumps (#762) 2023-09-05 09:07:44 -04:00
Nathaniel Simard c484999d54
Fix CI (#766) 2023-09-04 20:55:31 -04:00
Arvid Hammarlund b58af4a4a3
Cross Entropy with label smoothing and weights. (#755) 2023-09-04 20:20:47 -04:00
Nathaniel Simard 36446e8d35
fix: rnn mut (#764) 2023-09-04 10:09:15 -04:00
Damien Elmes 5b97f1a54b
Re-export some submodules and add feature flags for them (#759) 2023-09-04 09:13:52 -04:00
Damien Elmes ede02d5305
Exclude image crate when sqlite feature disabled (#758)
Not needed for anything but huggingface, which is unavailable when
sqlite is unavailable.
2023-09-03 09:01:17 -04:00
Damien Elmes 08e2ccbed3
Fix: log file creation could not be avoided (#754) 2023-09-03 08:50:48 -04:00
Louis Fortier-Dubois 56b8e29514
Doc/building blocks more (#745) 2023-09-02 13:09:03 -04:00
Justin Moore 06157d3cde
Perf/ndarray: Optimize `conv2d` operation (#747) 2023-09-02 11:34:58 -04:00
Damien Elmes a47d23c3dd
Add ability to interrupt training loop (#753) 2023-09-02 11:31:46 -04:00
Damien Elmes d80e0d1734
Add ui/metrics feature flags (#740) 2023-09-02 11:26:40 -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 419df3383a
powf and stabilize candle (#748) 2023-09-01 10:50:44 -04:00
Damien Elmes 520cea16df
Fix: burn-core/std depended on sqlite feature (#731) 2023-09-01 09:28:58 -04:00
Damien Elmes 5839806687
Convert some flaky tests to use approx comparisons (#746)
* Convert some flaky tests to use approx comparisons

Closes #741

* Mention approximate runtime in the scripts
2023-09-01 09:01:15 -04:00
Damien Elmes 3669d2a6d4
Migrate from log4rs to tracing (#739) 2023-08-31 21:07:26 -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