Basic_CNNs_TensorFlow2 icon indicating copy to clipboard operation
Basic_CNNs_TensorFlow2 copied to clipboard

A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).

Results 27 Basic_CNNs_TensorFlow2 issues
Sort by recently updated
recently updated
newest added

Hi brother, well-done. Can you create confusion_matrix function in evaluate.py if possible?

It may sound like a naive question, but before starting training, which path should I put my images in?

After each epoch I have te same values: Epoch: 0/2, train loss: nan, train accuracy: 0.10688, valid loss: nan, valid accuracy: 0.10680 regardless of the model. tensorflow 2.3.1 (install via...

How to write the code to assign different gradient weights to different classes of SparseCategoricalCrossentropy

都不用迁移学习的吗 硬训练啊 结果不理想

我在使用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 希望能得到您的回复