-
Benjamin Lefaudeux authored
* adding a parity unit test * code review, better testing, use torch defaults and check for the loss, log world size
3d02f052
* adding a parity unit test * code review, better testing, use torch defaults and check for the loss, log world size