feat(train, dist_train, plots): improved training, implemented distributed training, added plotting

b42a2f2

IvanBanny commited on Nov 10, 2024