search

jpuigcerver / PyLaia

A deep learning toolkit specialized for handwritten document analysis
MIT License
207 stars 43 forks source link

Fix tests for multiprocessing #47

Open starride-teklia opened 1 year ago

starride-teklia commented 1 year ago

The bump to Pytorch 1.13 broke some tests related to multiprocessing on CPU and GPU. We get the following errors:

On these tests:

tests/callbacks/learning_rate_test.py:# TODO: fix test with num_processes=2
tests/callbacks/training_timer_test.py:# TODO: fix test with num_processes=2
tests/loggers/epoch_csv_logger_test.py:# TODO: fix test with num_processes=2
tests/scripts/htr/decode_ctc_test.py:# TODO: fix test with nprocs=2
tests/scripts/htr/netout_test.py:# TODO: fix test with nprocs=2
tests/scripts/htr/train_ctc_test.py:# TODO: fix "ddp_cpu" mode
tests/scripts/htr/train_ctc_test.py:# TODO: fix "ddp" mode
tests/scripts/htr/train_ctc_test.py:# TODO: fix first assertion

I skipped the tests for now, but I need to investigate why we are getting this error and how to fix it.