Update balanced_positive_negative_sampler.py (#93)
use cuda version torch.randperm to avoid copy from gpu to cpu and a fatal bug in multi-thread cpu version
Showing
想要评论请 注册 或 登录
use cuda version torch.randperm to avoid copy from gpu to cpu and a fatal bug in multi-thread cpu version