eric-is-good / vit_wgangp Goto Github PK
View Code? Open in Web Editor NEWVITGAN 一个结合了 vision transformer 和 wgangp 的对抗生成网络，生成动漫头像，同时提供一个传统卷积网络的训练好的模型，仅仅 20 M.
vit_wgangp's People
Contributors
Stargazers
Watchers
Forkers
yanewwgvvit_wgangp's Issues
您好，有代码的使用文档吗？
本科阶段并没有进行过相应的学习，导师和师兄也不是该方向的，我该具体如何运行和调试该代码呢？
您好，为什么我的训练结果是噪声？
我是用vit和wgangp，训练后是噪声，麻烦问一下是需要调节哪些参数吗还是数据集的原因？训练100次会越来越清晰吗？前几次全是马赛克噪声。
出现报错RuntimeError: mat1 and mat2 shapes cannot be multiplied (57x215296 and 256x100)，不知道是什么问题，向您请教。
您好，感谢您分享的代码！
我在使用自己的训练集运行的时候，出现一下问题，请教您如何解决，谢谢。
100%|██████████| 57/57 [00:00<00:00, 491.31it/s]
0%| | 0/100 [00:00<?, ?it/s]
Traceback (most recent call last):
File "C:\code\WGAN_GP\train_for_vae.py", line 367, in
trainer.train(num_epochs=100,
File "C:\code\WGAN_GP\train_for_vae.py", line 151, in train
recon_loss, kl_diverge = self.compute_batch(batch)
File "C:\code\WGAN_GP\train_for_vae.py", line 196, in compute_batch
outputs, mu, log_var = self.model(images)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "C:\code\WGAN_GP\train_for_vae.py", line 88, in forward
mu, log_var = self.encoder(x)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "C:\code\WGAN_GP\model\vaemodel.py", line 99, in forward
mu, log_var = self.mu(outputs), self.log_var(outputs)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\linear.py", line 103, in forward
return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (57x215296 and 256x100)
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.