PaddleGAN
PaddleGAN copied to clipboard
咨询first-order-demo的相关问题
-
applications/tools/first-order-demo.py
里面,有一句paddle.set_device('cpu')
,从代码上看是仅在用户设置了--cpu
的时候,才会执行这句,进而用cpu计算。那么默认的是啥?换句话说,我猜测有以下两种可能,哪个是正确的呢(或者两个都不对)? (1)如果用户的设备是gpu、npu或者xpu等等异构设备,会自动寻找并且使用。(没找到相关代码) (2)默认的就是gpu,如果要用npu或者xpu设备,需要显式调用set_device
。(没找到相关代码) -
ppgan/apps/first_order_predictor.py
的第255行,有个操作是直接把driving_video
转nparray
紧接着用paddle.to_tensor
,那么当驱动视频很大,无法完全装进设备的显存,咋办?从代码上看,我猜可以先转nparray
,然后切成一堆小段,分别to_tensor
,每一段自己算自己的,最后把所有的输出都append
在一起,这样可行吗?