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

245 Commits

Author SHA1 Message Date
Nathaniel Simard eea5a263bf
Feat/adam optimizer (#140) 2022-12-30 15:02:43 -05:00
Visual 567adfb93e
refactor: fix all clippy warnings (#137) 2022-12-25 11:22:25 -05:00
Visual 85f98b9d54
refactor, feat: clean Cargo.toml files, upgrade tch to 0.10 (#131) 
* Clean Cargo.toml files, upgrade tch to 0.10

* Add pull_request hook to test.yml workflow
 2022-12-25 10:36:23 -05:00
Nathaniel Simard 3a9dfe6097
feat: cross entropy loss (#130) 2022-12-25 10:10:22 -05:00
Nathaniel Simard 3a91c2c48e
Feat/multi device (#128) 2022-12-20 18:01:58 -05:00
Nathaniel Simard a599eaed88
Feat/module visitor (#127) 2022-12-17 14:25:36 -05:00
Nathaniel Simard d9592411c2
Feat/text generation example (#126) 2022-12-16 19:23:51 -05:00
Nathaniel Simard eee90a5c9e
Example/text classification (#123) 2022-12-02 17:42:49 -05:00
Nathaniel Simard 46d06f0c90
feat: module init (#117) 2022-11-25 22:02:26 -05:00
Nathaniel Simard 4abc281dae
Refactor/forward pass (#116) 2022-11-25 21:14:44 -05:00
Nathaniel Simard acb14adc29
Fix/named tensor (#115) 2022-11-25 19:30:35 -05:00
Nathaniel Simard e0e787f87d
Experimental/named tensor (#113) 2022-11-23 19:05:46 -05:00
Nathaniel Simard c4c739d91b
doc: update readme (#112) 2022-11-20 19:41:55 -05:00
Nathaniel Simard dda067e79b
Refactor/backend autodiff (#106) 2022-11-19 12:37:06 -05:00
Nathaniel Simard d45d674a04
Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
Nathaniel Simard ab51c22a55
Refactor/extract tch backend (#103) 2022-11-15 21:06:40 -05:00
Nathaniel Simard 3da122db09
Add cuda gpu example + doc (#91) 2022-11-09 21:32:51 -05:00
Nathan Henrie 947ed00301
Fix `cargo run --example mnist` (#90) 2022-11-09 20:45:58 -05:00
Nathaniel Simard c5213b6c32
Fix: mnist example (#85) 2022-11-07 17:58:50 -05:00
Nathaniel Simard 5b4855317b
Doc/improve example (#64) 2022-10-26 18:08:38 -04:00
Nathaniel Simard bd36023b06
Update projects (#29) 2022-09-04 14:22:56 -04:00
Nathaniel Simard 548a1be488
Refactor/burn tensor (#28) 2022-09-03 12:51:38 -04:00
Nathaniel Simard 1ff11e537a
Feat/save load (#27) 2022-09-01 20:31:45 -04:00
Nathaniel Simard 674e078a85
Feat/inner module (#26) 2022-08-30 18:05:42 -04:00
nathaniel 68e052513b refactor: trainer 2022-08-27 11:54:51 -04:00
nathaniel 2c4288d536 fix integer backend 2022-08-27 11:34:53 -04:00
nathaniel 9ed0b0bdcd refactor: multi-thread logger -> async logger 2022-08-26 20:59:21 -04:00
nathaniel 05eb2f9cfd feat: async logging 2022-08-26 20:55:42 -04:00
nathaniel 32d81e4fdd refactor: dataloader 2022-08-26 20:25:19 -04:00
nathaniel 9ed252755c chore: cleanup 2022-08-26 18:38:16 -04:00
Nathaniel Simard 8da87b376d
Feat/use index assign for cat (#24) 2022-08-26 15:41:29 -04:00
Nathaniel Simard 508cfd2b5a
Refactor/metric (#23) 2022-08-23 16:39:30 -04:00
Nathaniel Simard d62f2b09ad
Feat/argmax (#22) 2022-08-23 13:57:30 -04:00
Nathaniel Simard 0aa486fc0c
Feat/logger (#21) 2022-08-23 11:47:21 -04:00
nathaniel bad608e72c feat: supervised trainer 2022-08-22 19:54:33 -04:00
Nathaniel Simard 7675ca0776
Feat/mix precision (#20) 2022-08-22 16:03:53 -04:00
Nathaniel Simard 5560bae26a
Feat/elem conversion (#18) 2022-08-22 13:27:15 -04:00
Nathaniel Simard 0f6b50fc10
Feat/dataloader (#17) 2022-08-22 11:21:22 -04:00
Nathaniel Simard 1b8b1e3843
Feat/dataset huggignface (#16) 2022-08-21 17:20:45 -04:00
nathaniel 39be42b2c6 feat: basic setup 2022-08-20 20:20:56 -04:00
Nathaniel Simard d00b74b527
Feat/tensor concrete type (#9) 2022-07-31 12:06:25 -04:00
nathaniel b0f0d7e378 refactor: create Tensor + Element trait 2022-07-27 10:20:21 -04:00
nathaniel 51cb331aef refactor: imports 2022-07-27 09:37:56 -04:00
nathaniel 3ba6c69875 feat: full support for ndarray 2022-07-26 18:01:29 -04:00
nathaniel 61b67b44ff doc: add a simple example 2022-07-26 10:04:13 -04:00