Congratulate for your wonderful work. I have a question. Can we train or test this code on windows with CPU? Thanks.