Commit Graph

735 Commits

Author SHA1 Message Date
Nathaniel Simard 34ad49d1b2
feat: support fields that are not param in module (#32) 2022-09-10 16:35:17 -04:00
Nathaniel Simard d0b581a919
Add docs to the Tensor struct (#31) 2022-09-09 18:10:25 -04:00
Nathaniel Simard 823f28c2b9
Fix README (#30) 2022-09-04 21:29:34 -04:00
nathaniel 5f292423d2 Fix deploy script 2022-09-04 14:27:27 -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 7988000dad Generate and Load from State 2022-08-31 19:39:48 -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 f8ab29bba6
feat: add comparison (#19) 2022-08-22 14:27:23 -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 6ef4653d21 feat: add burn-dataset 2022-08-21 09:42:42 -04:00
nathaniel 39be42b2c6 feat: basic setup 2022-08-20 20:20:56 -04:00
nathaniel 0f38096abe feat: cross entropy loss 2022-08-20 17:27:04 -04:00
Nathaniel Simard 6642a49d84
Feat/softmax (#15) 2022-08-20 17:13:55 -04:00
nathaniel 7e78aa09ea Feat: add div ops 2022-08-20 12:15:58 -04:00
Nathaniel Simard b2912f483a
Feat/more ops (#14) 2022-08-20 11:20:25 -04:00
nathaniel 71d37f1ec6 feat: relu activation function 2022-08-10 19:09:43 -04:00
nathaniel adf56ed9e6 refactor: fix tensor generics order 2022-08-09 12:40:51 -04:00
nathaniel 77d7c84ab3 Fix element 2022-08-08 19:18:54 -04:00
nathaniel 127b7cb60b wip 2022-08-08 19:18:54 -04:00
nathaniel 5d6c866198 feat: implement mask fill ops 2022-08-08 08:23:55 -04:00
Nathaniel Simard c261398814
Feat/more ops (#11) 2022-08-07 17:28:08 -04:00
Nathaniel Simard 0059c611f5
Feat/module definition (#10) 2022-08-05 21:14:44 -04:00
nathaniel e2f3a6fa10 feat: add backend infos 2022-07-31 14:01:42 -04:00
Nathaniel Simard d00b74b527
Feat/tensor concrete type (#9) 2022-07-31 12:06:25 -04:00
nathaniel 2872b488ef feat: remove some where for ndarray 2022-07-28 14:54:03 -04:00
nathaniel 5f9e0d1bac ci: publish to crates.io 2022-07-27 16:54:34 -04:00
nathaniel bfbdfbeb77 doc: update README 2022-07-27 16:15:48 -04:00
nathaniel 305438a476 Merge branch 'main' of github.com:nathanielsimard/burn 2022-07-27 15:45:10 -04:00
nathaniel ae21cfa34d doc: add burn-tensor Readme 2022-07-27 15:44:41 -04:00
Nathaniel Simard 43f2d58a82
Update README.md 2022-07-27 14:37:48 -04:00
nathaniel 746cf79c4b ci: setup ci 2022-07-27 12:26:54 -04:00
nathaniel 7698dd7684 refactor: cleanup backends 2022-07-27 11:56:30 -04:00
nathaniel fbf5fe3dda refactor: move zeros + ones impl 2022-07-27 11:42:50 -04:00
nathaniel acbbcc3781 feat: add creation trait 2022-07-27 11:24:41 -04:00