lucasdelimanogueira
|
77cabe41a2
|
fix cross entropy loss cuda
|
2024-05-23 12:19:32 -03:00 |
|
lucasdelimanogueira
|
8b19ac38c9
|
fix send module to device
|
2024-05-23 10:20:08 -03:00 |
|
lucasdelimanogueira
|
2c2569d482
|
operations unit tests
|
2024-05-07 11:52:56 -03:00 |
|
lucasdelimanogueira
|
1ccf9a6c8d
|
some tests test.py
|
2024-05-07 01:00:08 -03:00 |
|
lucasdelimanogueira
|
ffe10e48c5
|
Fix grad update step! loss decreasing
|
2024-05-07 00:10:25 -03:00 |
|
lucasdelimanogueira
|
b034d86ec3
|
Fix recursion backward increasing a lot
|
2024-05-06 22:58:37 -03:00 |
|
lucasdelimanogueira
|
de6f2af637
|
fix upgrade grads reference to parameter
|
2024-05-06 20:40:06 -03:00 |
|
lucasdelimanogueira
|
e5819a4121
|
nn module v1
|
2024-05-06 19:47:27 -03:00 |
|
lucasdelimanogueira
|
1e00930c4e
|
Fix nn.Module linear
|
2024-05-06 13:23:22 -03:00 |
|