lucasdelimanogueira
|
10a88fe8ca
|
unit test dataloader and distributedsampler
|
2024-05-25 11:55:59 -03:00 |
|
lucasdelimanogueira
|
f13d22eb91
|
fix send to device unittest
|
2024-05-23 10:04:44 -03:00 |
|
lucasdelimanogueira
|
5946e6e1bd
|
fix unittest test nn cuda
|
2024-05-23 03:01:27 -03:00 |
|
lucasdelimanogueira
|
d530f79a82
|
cross entropy loss
|
2024-05-21 12:16:12 -03:00 |
|
lucasdelimanogueira
|
a39bd59a3f
|
softmax operation
|
2024-05-21 01:22:38 -03:00 |
|
lucasdelimanogueira
|
5e72fe57d3
|
softmax v1 need fix some erros yet
|
2024-05-20 20:16:41 -03:00 |
|
lucasdelimanogueira
|
c11cfd7f22
|
fix transpose cuda
|
2024-05-16 19:10:22 -03:00 |
|
lucasdelimanogueira
|
85d8e74ad1
|
fix cuda add broadcasted
|
2024-05-16 18:06:50 -03:00 |
|
lucasdelimanogueira
|
e6104e43c9
|
small changes
|
2024-05-10 20:12:15 -03:00 |
|
lucasdelimanogueira
|
6b08c10300
|
Cosine and sine operations with autograd
|
2024-05-09 22:32:36 -03:00 |
|
lucasdelimanogueira
|
7e3100f796
|
unit test nn module mse loss and sigmoid
|
2024-05-07 17:44:59 -03:00 |
|