AmoebaNet-B is a convolutional neural network for visual recognition. Trained with GPipe across four accelerators in four stages, its 557 million parameters reached 84.4% top-1 accuracy on ImageNet-2012, then a single-model benchmark record. section 6.3.2
Its size also supported transfer learning: the ImageNet-trained model matched or exceeded results across several vision datasets. Convolutional layers had uneven computational costs, however, making the model harder to partition efficiently; on eight GPUs, the observed speedup was 3.5-fold. section 6.3.2
CIFAR-100 transfer
Fine-tuning the ImageNet-trained model reduced the CIFAR-100 error rate from 10.7% to 8.7%. section 6.3.2
