Comments (2)
HolmesShuan
commented on July 29, 2024
FYI.
Our model is trained on two NVIDIA Tesla P100 GPUs and two NVIDIA GeForce RTX 3090 GPUs.
FP16 seems to be a good option if you encounter OOM.
from stytr-2.
Jamie-Cheung
commented on July 29, 2024
font{
line-height: 1.6;
}
ul,ol{
padding-left: 20px;
list-style-position: inside;
}
Thank you for your time. I want to know whether your code can be trained on 2*3090. I failed to train it on my computer, Ubuntu18 2*3090.
Error:RuntimeError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 24.00 GiB total capacity; 21.24 GiB already allocated; 210.21 MiB free; 21.30 GiB reserved in total by PyTorch)
***@***.***
在2022年3月12日 ***@***.***> 写道：
Closed #6.
—Reply to this email directly, view it on GitHub, or unsubscribe.Triage notifications on the go with GitHub Mobile for iOS or Android.
You are receiving this because you authored the thread.Message ID: ***@***.***>
from stytr-2.
Related Issues (20)
- dataset HOT 1
- PatchEmbed源码和论文图中好像操作方法好像不太一样 HOT 4
- 作者您好！对于位置编码有一些疑问
- About the patch partition. HOT 1
- inference parameters
- Arbitrary output size instead of square HOT 3
- Request for elaboration regarding the Dataset HOT 2
- content leak HOT 1
- Error in running "train.py"
- 没有结果生成 HOT 5
- AttributeError: 'tuple' object has no attribute 'cpu' HOT 1
- Hello, thank you for your great work.
- Hi,please help me HOT 4
- About the metric score of StyTr2:Image Style Transfer with Transformers
- 关于代码中构建的transformer的问题
- running train.py get wrong
- 0
- 非常感谢你的论文做出的贡献
- I replace the code with yours, but I got the output as this HOT 2
- Loss weight different values different from the paper
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 stytr-2.