c52c49785d
Add linear learning rate scheduler ( #1443 )
2024-03-12 13:04:12 -04:00
0138e16af6
Add Enum module support in PyTorchFileRecorder ( #1436 )
...
* Add Enum module support in PyTorchFileRecorder
Fixes #1431
* Fix wording/typos per PR feedback
2024-03-11 11:21:01 -05:00
c7d4c23f97
Support for non-contiguous indexes in PyTorchFileRecorder keys ( #1432 )
...
* Fix non-contiguous indexes
* Update pytorch-model.md
* Simplify multiple forwards
2024-03-07 13:40:57 -06:00
b12646de0a
Truncate debug display for NestedValue ( #1428 )
...
* Truncate debug display for NestedValue
* Fix failing tests
2024-03-07 08:06:31 -05:00
545444c02a
PyTorchFileRecord print debug option ( #1425 )
...
* Add debug print option to PyTorchFileRecorder
* Updated documentation and improved print output
* Improve print wording
* Updated per PR feedback
2024-03-06 16:11:37 -06:00
d43a0b3f90
Add is_close and all_close tensor operators ( #1389 )
...
* Add is_close and all_close tensor operators
* Fix broken build issues
* Fix the table
* Add tests to candle
2024-03-01 15:37:14 -06:00
688958ee74
Enhance PyTorchRecorder to pass top-level key to extract state_dict ( #1300 )
...
* Enhance PyTorchRecorder to pass top level key to extract state_dict
This is needed for Whisper weight pt files.
* Fix missing hyphens
* Move top-level-key test under crates
* Add sub-crates as members of workspace
* Update Cargo.lock
* Add accidentally omitted line during merge
2024-02-29 12:57:27 -06:00
330552afb4
docs(book-&-examples): modify book and examples with new `prelude` module ( #1372 )
2024-02-28 13:25:25 -05:00
8e23057c6b
Feature Addition: PRelu Module ( #1328 )
2024-02-24 10:24:22 -05:00
1da47c9bf1
feat: add prelude module for convenience ( #1335 )
2024-02-24 10:17:30 -05:00
27f2095bcd
Implement Instance Normalization ( #1321 )
...
* config
* rename as instances, otherwise won't work
* refactor
* InstanceNormConfig
* remove unused var
* forward
* rename
* based on gn
* unit tests
* fix tests
* update doc
* update onnx doc
* renaming method
* add comment
---------
Co-authored-by: VungleTienan <tienan.liu@vungle.com>
2024-02-23 23:31:43 -06:00
08302e38fc
Fix broken test and run-checks script ( #1347 )
2024-02-23 10:06:51 -05:00
c86db83fa9
Add support for Any, All operations to Tensor ( #1342 )
...
* add any, all op implementation for all tensor types
* add op to burn-book
* fix formatting
* refactor tensor operations from numeric to BaseOps.
* fix book doc
* comments fix and add more tests
2024-02-23 10:06:31 -05:00
d6e859330f
Pytorch message updates ( #1344 )
...
* Update pytorch-model.md
* Update error.rs
2024-02-22 12:12:50 -06:00
bff4961426
Add enum module support ( #1337 )
2024-02-21 17:03:34 -05:00
4427768570
[refactor] Move burn crates to their own crates directory ( #1336 )
2024-02-20 13:57:55 -05:00
Kyle Chen