Comments (12)
WkEEn
commented on July 20, 2024
from deep-learning-for-image-processing.
WZMIAOMIAO
commented on July 20, 2024
你训练的是什么模型
from deep-learning-for-image-processing.
WkEEn
commented on July 20, 2024
vgg
环境：pytorch1.3.1+cu92+python3.7
from deep-learning-for-image-processing.
WkEEn
commented on July 20, 2024
from deep-learning-for-image-processing.
WZMIAOMIAO
commented on July 20, 2024
我初步怀疑是你的GPU内存不够，因为VGG 网络模型参数太大了，你可以尝试一下步骤来排除问题：
1）你使用alexnet的训练脚本调用GPU进行训练（如果没问题说明你的GPU环境是正常的）
2）如果第一步没问题再使用VGG的训练脚本，将batchsize设置成1进行训练（如果能够正常训练说明是内存不足的问题）
from deep-learning-for-image-processing.
WkEEn
commented on July 20, 2024
我尝试自己搭建的简单网络，还会报这样的错，请问是GPU的原因吗？
@WZMIAOMIAO
from deep-learning-for-image-processing.
WZMIAOMIAO
commented on July 20, 2024
你GPU的内存多大，也有可能是你GPU环境配置的问题。
from deep-learning-for-image-processing.
WkEEn
commented on July 20, 2024
from deep-learning-for-image-processing.
WkEEn
commented on July 20, 2024
from deep-learning-for-image-processing.
WZMIAOMIAO
commented on July 20, 2024
你的GPU配置挺好的，应该是你GPU环境配置问题。
建议试下这个配置pytorch1.3 + CUDA10.1 + cuDNN7.6
我的所有代码都是在这个环境下跑的。
from deep-learning-for-image-processing.
roseif
commented on July 20, 2024
安装多一个环境安装最新的Pytorch试下，我的没有问题
from deep-learning-for-image-processing.
WkEEn
commented on July 20, 2024
ok，确实是版本问题
@roseif
from deep-learning-for-image-processing.
Related Issues (20)
- PPT放一个地方不好吗？ 一个个分开，整理也够累的 HOT 2
- FileNotFoundError: Found no valid file for the classes roses, sunflowers, tulips. Supported extensions are: .jpg, .jpeg, .png, .ppm, .bmp, .pgm, .tif, .tiff, .webp
- 在torch >= 2.0中移除了名为“torch._six”的模块 coco_eval.py文件可能需要一些修改 HOT 2
- 请问大家这是我训练10个epoch后的结果，可能是些什么原因出错了 HOT 1
- yolov3_spp的导入包错误
- 你好作者，关于EffiectentV2模型pytorch版本，预训练权重使用的是什么数据集进行预训练的呢？
- 针对于自己 的模型生成Grad-Cam
- 不理解这里的架构，是有问题吗
- swin_transformer export onnx : Process finished with exit code 136 (interrupted by signal 8: SIGFPE)
- FCN网络代码中，是否有pool3，pool4，pool5的特征融合过程呢
- 训练文件运行错误？
- 大佬vit那个项目的requirments.txt是啥呀，按照你readme中的要求下载了，遇到了很多版本不兼容的问题，球球了 HOT 1
- 在云上运行FCN网络的时候使用GPU进行训练会报这个错：FileNotFoundError: [Errno 2] No such file or directory srun: error: gpu03: task 0: Exited with exit code 1 HOT 5
- HRNet HOT 3
- HRNet相关问题
- 请教vit模型，百度网盘里面的权重是怎么得到的？自己重新训练的还是从官方实现的npz权重转换过来的？ HOT 1
- VGG模型分类后训练找不到文件 HOT 1
- pytorch版本问题 HOT 1
- 如何完整的将Faster-rcnn导出为ONNX？
- 关于置信度的调整
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 deep-learning-for-image-processing.