Question about training speed

[hzhz2020]

Dear Authors

Your work is exciting! I am trying out your code with the example you provides, python Train_CoMatch.py --n-labeled 40 --seed 1 I am running on one A100 gpu. I found that the speed is very slow, takes ~10 minutes for 1 epoch. Is this the expected behavior? I just want to make sure i am running the code correctly.

Thank you, and hope to hear back from you!