heha102
heha102
I met the same situation. Do you know how to solve it now?
https://github.com/Adamdad/ConsistentTeacher/issues/15 this issue solved the problem
The same problem.
参考https://github.com/microsoft/SoftTeacher/issues/109#issuecomment-976197003 修改semi_sampler.py的 211行, 将 sum(self.epoch_length) * self.samples_per_gpu,修改为sum(self.epoch_length)。这样是可以的。
I tried what @ttt0666 suggested, but I'm get a new error. I referred to this https://www.neusncp.com/user/blog?id=766 , added `os.environ['CUBLAS_WORKSPACE_CONFIG']=':4096:2'` at the top of tools/train.py, which successfully addressed the problem.