Comments (4)
If I had to guess, you're probably getting an exception when trying to load the FusedLeakyReLU?
We are not sure when we will have time to add CPU inference support, but you may find the following issue from rosinality helpful:
rosinality/stylegan2-pytorch#89
The commit linked in the issue above should be a good starting point, but some more modifications will most likely be needed.
from pixel2style2pixel.
Hi @justmaulik
This may be possible, but it may be challenging to make the changes to the code as ninja requires CUDA.
However, if you plan on only running inference (and don't have access to a GPU), maybe running on Google colab will suit your needs.
In case you missed it, we have a Google colab linked in the README that can help you get started.
from pixel2style2pixel.
I already tried on colab and also modified some of the code(few lines) where it required CUDA but it's more complex than just a few lines. and I can train it on GPU but I am planning to host inference as API so CPU is the only way the fits my budget.
Replacing ninja to support CPU inference is anywhere on the to-do list? :) or you could just point me in a direction what I need to do to run just the inference on CPU.
from pixel2style2pixel.
Thanks @yuval-alaluf
from pixel2style2pixel.
Related Issues (20)
- do a huggingface demo HOT 1
- How to train on paired data? HOT 1
- latent image editing HOT 1
- About celebs_seg_to_face HOT 1
- multiple GPUs HOT 4
- Using my own pretrained model from vanilla StyleGAN2 HOT 3
- Is it possible video2anime train? like this project? HOT 3
- A problem about how to get diverse images? HOT 1
- the output image become brown HOT 3
- Need help with running the code on CPU HOT 1
- loading pretrained weights, STACK_GLOBAL requires str HOT 1
- How to train pSp on Z+ space? HOT 2
- Is it possible to create output images in profile (side-on) perspective using sketch to face? HOT 1
- Single channel input with Moco loss not working. HOT 2
- Some error reporting problems encountered during operation HOT 1
- How to use other identity loss rather than Arcface or Moco HOT 2
- How to use psp for Face beautification
- loss jump problem HOT 1
- Retraining
- How can I use the style mixing to just change the hair color.
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 pixel2style2pixel.