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
|
e590f57555
|
implemented distributed sampler
|
2024-05-25 10:53:09 -03:00 |
|
lucasdelimanogueira
|
ffe10e48c5
|
Fix grad update step! loss decreasing
|
2024-05-07 00:10:25 -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 |
|