lucasdelimanogueira
|
e6d58121f1
|
fix memory usage
|
2024-06-05 20:24:09 -03:00 |
|
lucasdelimanogueira
|
e670559fe4
|
fix memory issues del tensors
|
2024-06-05 15:43:19 -03:00 |
|
lucasdelimanogueira
|
6701179c22
|
del tensor free memory
|
2024-06-05 14:31:48 -03:00 |
|
lucasdelimanogueira
|
64f08a8963
|
multigpu training support v0.0.5
|
2024-06-03 15:37:57 -03:00 |
|
lucasdelimanogueira
|
c42d334599
|
fix allreduce distributed not working after 2nd iteration
|
2024-06-03 15:14:42 -03:00 |
|
lucasdelimanogueira
|
e706b81007
|
fix broadcast and reduce gradients distributeddataparallel and tensor hooks
|
2024-06-03 11:40:05 -03:00 |
|
lucasdelimanogueira
|
9634ab12c5
|
Fix long time step() detach
|
2024-05-07 01:27:48 -03:00 |
|
lucasdelimanogueira
|
838af9cf4c
|
Fix long time step() detach
|
2024-05-07 01:19:12 -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 |
|