Giters

Adamdad / keras-YOLOv3-mobilenet

I transfer the backend of yolov3 into Mobilenetv1，VGG16，ResNet101 and ResNeXt101

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train the model with GPU?

LiuShaw opened this issue · comments

commented

I have use the "os.environ["CUDA_VISIBLE_DEVICES"] = "0", but the traing still use CPU. How to setting to use GPU?