Colab 中的并行计算

问题描述

我正在尝试进行一些并行计算(这是第一次!),但我真的不知道该怎么做,或者它是否会加快我的计算速度。

我在 Colab Notebook 中有一个神经网络,我必须将相同的小批量图像运行 N 次,以便进行一些 dropout 统计

这可能是一项非常简单的任务,但我不知道该怎么做。代码很简单:

for i in np.arange(iters):
   sample[i] = model(x)

或者类似的东西,你懂的。 问题是模型 (x) 消耗了相当多的时间,我真的需要并行执行。

另外,一个有点相关的问题:Colab 有多少个内核?迭代次数应该在 10.000 左右,所以可能太多了,不是吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)