Go to file
2024-09-26 13:45:53 +00:00
context_parallel.py ugly ass display of grid (to be changed) 2024-09-26 13:45:53 +00:00
data_parallel.py support CPU training through gloo backend 2024-09-26 10:27:20 +00:00
dataset.py rename parallel_context to process_group_manager 2024-09-25 13:33:20 +00:00
distributed_primtives.py support CPU training through gloo backend 2024-09-26 10:27:20 +00:00
generate.py rename parallel_context to process_group_manager 2024-09-25 13:33:20 +00:00
pipeline_parallel.py support CPU training through gloo backend 2024-09-26 10:27:20 +00:00
process_group_manager.py ugly ass display of grid (to be changed) 2024-09-26 13:45:53 +00:00
README.md Initial commit 2024-09-18 14:01:22 +02:00
requirements.txt add wandb support 2024-09-25 14:19:16 +00:00
train.py ugly ass display of grid (to be changed) 2024-09-26 13:45:53 +00:00
utils.py ugly ass display of grid (to be changed) 2024-09-26 13:45:53 +00:00

picotron