使用 Multi GPU 环境进行火炬训练 - torch training with Multi GPU enviroment
我正在尝试在多 gpu 环境中进行培训。 这是 model 代码 snn.Leaky 是一个用于实现 SNN 结构的模块,与 torch.nn 相结合,使 .network 像 RNN 一样工作。 链接在这里( https://snntorch.readthedocs.io/en/latest/r ...