Comments (6)
I‘ve added an option "--device" for CPU inference. Feel free if you have any questions.
from livespeechportraits.
@YuanxunLu thanks i still get this error when using --device cpu
Traceback (most recent call last):
File "demo.py", line 27, in
from funcs import utils
File "/content/LiveSpeechPortraits/funcs/utils.py", line 60, in
n_mel_channels=80, mel_fmin=90, mel_fmax=7600.0).cuda()
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 637, in cuda
return self._apply(lambda t: t.cuda(device))
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 573, in _apply
self._buffers[key] = fn(buf)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 637, in
return self._apply(lambda t: t.cuda(device))
File "/usr/local/lib/python3.7/dist-packages/torch/cuda/init.py", line 172, in _lazy_init
torch._C._cuda_init()
RuntimeError: No CUDA GPUs are available
from livespeechportraits.
Thanks for your feedback. I've modified the functions and please check the latest version.
from livespeechportraits.
@YuanxunLu thanks i am getting this error now
Traceback (most recent call last):
File "demo.py", line 130, in
Featopt = FeatureOptions().parse()
File "/content/LiveSpeechPortraits/options/base_options_audio2feature.py", line 158, in parse
torch.cuda.set_device(opt.gpu_ids[0])
File "/usr/local/lib/python3.7/dist-packages/torch/cuda/init.py", line 264, in set_device
torch._C._cuda_setDevice(device)
File "/usr/local/lib/python3.7/dist-packages/torch/cuda/init.py", line 172, in _lazy_init
torch._C._cuda_init()
RuntimeError: No CUDA GPUs are available
from livespeechportraits.
I‘ve commented these codes in the latest version. Check it if it has any questions.
from livespeechportraits.
@YuanxunLu
pls change these code and then cpu inference can work
FYI pytorch cpu version is required
1.lossess.py
# return current_sample.reshape(b, T, -1).cuda()
return current_sample.reshape(b, T, -1)
2.demo.py
if torch.cuda.is_available():
map_location=lambda storage, loc: storage.cuda()
else:
map_location='cpu'
APC_model.load_state_dict(torch.load(config['model_params']['APC']['ckp_path'],map_location=map_location), strict=False)
from livespeechportraits.
Related Issues (20)
- Personalized data generation HOT 3
- What is the meaning of implementing by C++? HOT 1
- 候选照片,一共四张,是基于什么逻辑进行选择的?
- how can i use it in real time? HOT 1
- Does anyone implement the training code of this project? HOT 1
- How to run demo in "Real-time" HOT 1
- 模型得到的矩阵值可以和ARkit进行映射吗?
- RuntimeError: Found no NVIDIA driver on your system.
- Great project, where does the author achieve real-time performance? HOT 2
- 如何生成自己的模型。从哪里导入我的视频素材生成我自己的模型。
- How to train these models in custom dataset? Any documentation? HOT 1
- What tool did you use to create a sketch from a face image, in case i want to train the image to image transition model?
- 73 facial landmarks HOT 1
- FileNotFoundError: [Errno 2] No such file or directory: './data/May\\mean_pts3d.npy' HOT 1
- 数字人技术交流群请联系VX:metahuman668
- GMMLogLoss for training audio2headpose
- training data download
- Is the Released Models Trained on Whole Video Clip?
- code for data processing, training HOT 2
- REAL TIME 哪里去了?不是说好可以根据音频流来实时输出吗? HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from livespeechportraits.