Closed Gong-air closed 1 month ago
S2等没有实现pin_memory的vendor,需要在框架层对pin_memory设置为False,是不方便的。 因此在DIPU层对vendor进行判断,强行设置pin_memory为False,并打印相关LOG。
S2等没有实现pin_memory的vendor,需要在框架层对pin_memory设置为False,是不方便的。 因此在DIPU层对vendor进行判断,强行设置pin_memory为False,并打印相关LOG。