lucasdelimanogueira
|
e670559fe4
|
fix memory issues del tensors
|
2024-06-05 15:43:19 -03:00 |
|
lucasdelimanogueira
|
64f08a8963
|
multigpu training support v0.0.5
|
2024-06-03 15:37:57 -03:00 |
|
lucasdelimanogueira
|
38a860eba1
|
fix send to other devices and compile on cloud
|
2024-05-29 17:48:37 -03:00 |
|
lucasdelimanogueira
|
10a88fe8ca
|
unit test dataloader and distributedsampler
|
2024-05-25 11:55:59 -03:00 |
|
lucasdelimanogueira
|
8c5abb226c
|
version 0.0.4 cuda running neural network
|
2024-05-23 12:28:17 -03:00 |
|
lucasdelimanogueira
|
40a6bccab4
|
vision tranform, readme and example training
|
2024-05-22 14:25:02 -03:00 |
|
lucasdelimanogueira
|
0972a0873f
|
small fix crossentropy
|
2024-05-22 10:34:16 -03:00 |
|
lucasdelimanogueira
|
c808a49a30
|
dataset, dataloader and mnist
|
2024-05-18 17:59:14 -03:00 |
|
lucasdelimanogueira
|
418bb20b29
|
fix add broadcasted backward grad wrong values
|
2024-05-18 15:08:31 -03:00 |
|
lucasdelimanogueira
|
eb49d8cae3
|
fixed sum axis with keepdim
|
2024-05-17 20:12:57 -03:00 |
|
lucasdelimanogueira
|
5dd3f3d202
|
implementing keepdim sum
|
2024-05-17 19:08:09 -03:00 |
|
lucasdelimanogueira
|
e6104e43c9
|
small changes
|
2024-05-10 20:12:15 -03:00 |
|
lucasdelimanogueira
|
68bee8b457
|
add example training
|
2024-05-10 02:01:09 -03:00 |
|