search

calmiLovesAI / Basic_CNNs_TensorFlow2

A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).
MIT License
526 stars 178 forks source link

run error in regnet with GPU #27

Open tongshiyuan opened 3 years ago

tongshiyuan commented 3 years ago

我在使用GPU进行RegNet训练的时候，报错了

UnimplementedError:  Fused conv implementation does not support grouped convolutions for now.
     [[node reg_net/any_stage/res_bottleneck_block/bottleneck_transform/conv2d_3/BiasAdd (defined at /filepath/anynet.py:119) ]] [Op:__inference_train_function_10336]

Function call stack:
train_function

但是CPU下是正常的，我检查了一下代码，GPU和CPU模式下，get_group_conv是不一样的，但我不知道怎么debug 希望能得到您的回复

calmiLovesAI commented 2 years ago

根据实际情况更改配置文件中的这一项：DEVICE = "gpu"就行