ptrblck:


Use local variables instead of global: self.k instead of k (same for n_channel ).
Use multiple workers ( num_workers > 0 ) in your DataLoader .



These worked, in addition to putting my code in a dedicated main function and calling it. Thank you so much for your help. However, I am still not able to see the performance on compute_0 and compute_1 tabs on Task Manager. That’s okay for now, I think, since I do some improvement in speed.