search

aqibsaeed / Genetic-CNN

CNN architecture exploration using Genetic Algorithm
Apache License 2.0
215 stars 98 forks source link

Update GeneticCNN.py #10

Closed DarkoNedic closed 5 years ago

DarkoNedic commented 5 years ago

Regarding https://github.com/aqibsaeed/Genetic-CNN/pull/8#issuecomment-513038337 This fixes the shape of the tensors per stage after the pooling operation. The tensors should have a square shape now.