Comments (5)
Try this simple command to test if the pytorch and cuda are installed successfully:
import torch
print(torch.__version__)
print(torch.version.cuda)
Did you train your own "checkpoint_512_celeba-hq.pt" since that it needs so much GPU memory as comment in https://github.com/imlixinyang/HiSD/issues/12#issuecomment-840404898 (i.e. at least 4x Tesla V100)?
The released checkpoint supports to use 512x512 image as input since it will automatically resize the image into corresponding resolution.
from hisd.
Sorry, but I can't see what the content is in this url https://github.com/imlixinyang/HiSD/issues/12#issuecomment-840404898
It said "No results matched your search."
The result is that
torch.version = 1.0.1.post2
torch.version.cuda = 10.0.130
Now, I don't know that just only checkpoint_512 can output 512x512 image ?
Another question is where can I config the output image size ?
Thank you for your reply.
from hisd.
Only checkpoint_512 which is trained with 512 resolution can output 512x512 image directly. If you want to use 256x256 checkpoint to output 512x512 image, you should upsample the output in the test code.
Please ensure the original test code can run successfully first and then try your own modification.
from hisd.
Traceback (most recent call last):
File "core/train.py", line 64, in
G_adv, G_sty, G_rec, D_adv = trainer.update(x, y, i, j, j_trg)
File "/home/HiSD/core/trainer.py", line 139, in update
x_trg, x_cyc, s, s_trg = self.models((x, y, i, j, j_trg), mode='gen')
File "/home/anaconda3/envs/HiSD/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/HiSD/core/trainer.py", line 30, in forward
return self.gen_losses(*args)
File "/home/HiSD/core/trainer.py", line 40, in gen_losses
e = self.gen.encode(x)
File "/home/HiSD/core/networks.py", line 126, in encode
e = self.encoder(x)
File "/home/anaconda3/envs/HiSD/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/anaconda3/envs/HiSD/lib/python3.6/site-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/home/anaconda3/envs/HiSD/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/anaconda3/envs/HiSD/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 320, in forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
I run the origin test code which produce above message.
I have no idea.
from hisd.
It looks like that the reason is the python environment or packages. I recommend you to use conda environment (e.g., anaconda) and reinstall the cudatoolkit and pytorch following the repo.
from hisd.
Related Issues (20)
- About Mutil-Gpus for training? HOT 7
- How many tags can this project train at the same time? HOT 3
- Data imbalance HOT 1
- 作者您好,想请问一下关于discriminator的问题 HOT 6
- 请求RaFD数据集 HOT 4
- About beard model HOT 14
- A few questions about training tricks HOT 6
- 请教一下您的生成器和鉴别器模型是基于哪篇文章的 HOT 1
- 您好,请教一下论文中实验部分的一些问题,望指点 HOT 5
- 如何处理检查点文件? HOT 2
- 如何设置custom datasets? HOT 6
- 可以用m1芯片运行吗? HOT 3
- 更换数据集 HOT 2
- 关于生成图片模糊的问题 HOT 2
- Multi-tag task HOT 5
- 使用AFHQ数据集训练模型 HOT 13
- 求一个RaFD或者emotionnet数据集,想在celebA上做一下联合训练,有老哥有吗?我RaFD一直没申请到,emotionnet在网上下载的好像也是有问题的。
- 请教关于生成结果图像质量下降等问题 HOT 3
- 关于更高分辨率模型的训练问题 HOT 1
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 hisd.