lucasdelimanogueira
|
53983be40a
|
v0.0.7 improve memory usage
|
2024-06-05 21:07:07 -03:00 |
|
lucasdelimanogueira
|
64f08a8963
|
multigpu training support v0.0.5
|
2024-06-03 15:37:57 -03:00 |
|
lucasdelimanogueira
|
e590f57555
|
implemented distributed sampler
|
2024-05-25 10:53:09 -03:00 |
|
lucasdelimanogueira
|
56fff47001
|
fix import data module
|
2024-05-25 09:55:02 -03:00 |
|
lucasdelimanogueira
|
55db13836b
|
broadcast allreduce nccl
|
2024-05-24 18:19:39 -03:00 |
|
lucasdelimanogueira
|
55fc1ac020
|
version 0.0.4 cuda running neural network
|
2024-05-23 12:28:46 -03:00 |
|
lucasdelimanogueira
|
b72c16f110
|
version 0.0.3
|
2024-05-22 14:29:18 -03:00 |
|
lucasdelimanogueira
|
40a6bccab4
|
vision tranform, readme and example training
|
2024-05-22 14:25:02 -03:00 |
|
lucasdelimanogueira
|
c808a49a30
|
dataset, dataloader and mnist
|
2024-05-18 17:59:14 -03:00 |
|
lucasdelimanogueira
|
a846d3aa3e
|
Small fix for PiPy
|
2024-05-11 01:45:59 -03:00 |
|
lucasdelimanogueira
|
7e3100f796
|
unit test nn module mse loss and sigmoid
|
2024-05-07 17:44:59 -03:00 |
|
lucasdelimanogueira
|
e5819a4121
|
nn module v1
|
2024-05-06 19:47:27 -03:00 |
|
lucasdelimanogueira
|
8366245813
|
sub, elementmul, pow cuda operations
|
2024-04-30 00:31:46 -03:00 |
|