请教:
训练lda2vec的时候
gpu_id = int(os.getenv('CUDA_GPU', '0'))
cuda.get_device(gpu_id).use()
print("Using GPU " + str(gpu_id))
训练几十个epoch以后cupy.cuda.memory.OutOfMemoryError,然后又重头开始训练
data = prepare_topics(cuda.to_cpu(model.mixture.weights.W.data).copy(),
cuda.to_cpu(model.mixture.factors.W.data).copy(),
cuda.to_cpu(model.sampler.W.data).copy(),
words)