|
__pycache__
|
implemented mpi for distributed run
|
2024-05-24 15:12:53 -03:00 |
|
autograd
|
fix cross entropy loss cuda
|
2024-05-23 12:19:32 -03:00 |
|
csrc
|
fix import distributed
|
2024-05-24 19:55:19 -03:00 |
|
distributed
|
fix distributed allreduce calling python
|
2024-05-24 20:16:08 -03:00 |
|
nn
|
version 0.0.4 cuda running neural network
|
2024-05-23 12:28:17 -03:00 |
|
norchvision
|
vision tranform, readme and example training
|
2024-05-22 14:25:02 -03:00 |
|
optim
|
Fix long time step() detach
|
2024-05-07 01:27:48 -03:00 |
|
utils
|
dataset, dataloader and mnist
|
2024-05-18 17:59:14 -03:00 |
|
__init__.py
|
broadcast allreduce nccl
|
2024-05-24 18:19:39 -03:00 |
|
libtensor.so
|
fix import distributed
|
2024-05-24 19:55:19 -03:00 |
|
tensor.py
|
first commit distributed module
|
2024-05-23 19:15:30 -03:00 |
