Talos 多 GPU 功能

Talos multi gpu feature

我正在尝试 运行 Talos 超参数搜索我的 CNN。拥有 6 个 GPU 可以 运行 更快地进行实验,Talos 功能 multi_gpu 似乎很方便。

my_new_model = multi_gpu(my_new_model, gpus=6)
my_new_model.compile(...)

需要在 model.compile 之前调用函数。

但是好像还是只有一个job是同时执行的。 我希望同时处理具有 6 条可能路径的实验。现在还在一个接一个工作。

有什么想法吗?

multi_gpu 用于在一台机器上的多个 GPU 上执行单个任务 运行。

根据您在数据集、体系结构、参数等方面所做的工作,这可以显着减少实验时间。尽管仍然执行排列 "one by one",但每个排列的执行时间更短。