修改多线程的tensor方式为file_system:
torch.multiprocessing.set_sharing_strategy(‘file_system’)
默认方式为file_descriptor ,受限于open files数量
实测有效
修改多线程的tensor方式为file_system:
torch.multiprocessing.set_sharing_strategy(‘file_system’)
默认方式为file_descriptor ,受限于open files数量
实测有效