Skip to content

当服务器有4张卡的时候就训练不了了,准确来说是96/GPU数要是16的倍数。因为代码默认跑满四张卡,tensor的size不符合后面的处理。 #2

Description

@Li-Gui-233

可以将network_train_meta_learning.py文件的第70行代码修改为self.mymodel = torch.nn.DataParallel(self.mymodel, device_ids=[0, 1, 2])

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions