search

songdejia / DFL-CNN

This is a pytorch re-implementation of Learning a Discriminative Filter Bank Within a CNN for Fine-Grained Recognition
MIT License
267 stars 52 forks source link

请问单张1080ti可以跑这个网络吗？ #7

Open yuqiu1233 opened 5 years ago

yuqiu1233 commented 5 years ago

您好，请问单张1080ti可以跑这个网络吗，我看您用的是四张泰坦v，我应该如何修改参数，才能满足单张1080ti的显存使用？

wangkangnian commented 5 years ago

修改run.sh里面的 --gpu 参数为1

yuqiu1233 commented 5 years ago

我修改过，运行时候发现电脑异常卡顿，没一会就自动关机了，您知道是怎么回事吗？@wangkangnian

wangkangnian commented 5 years ago
  1. run.sh 写法有问题，你可以吧你的run.sh 发出来看下
  2. 电脑电源不够用
yuqiu1233 commented 5 years ago

@wangkangnian 谢谢你，我现在不在实验室。一会发出来