Hi,
Thank you for a very nice work.
I have noticed that after the training is over, the multiple processes aren't terminated.
Lines 370-372 in model.py:
for p in jobs:
p.join()
p.terminate()
I have "solved" the issue by masking the p.join() line but I guess this is not the optimal solution.
Can fix the issue?