howellyoung-s

Use DistributedShampoo but get unexpected error

2

**My system configure** Docker Image: tensorflow/tensorflow:latest-gpu-py3 Ubuntu 18.04 GCC 7.5.0 Tensorflow 2.1.0 CUDA 10.1 Python 3.6 Lingvo: 8eb904be3bc47e541f24a64a1a86751808b1bfe8 Commits on Apr 23, 2020 **Problem** I install lingvo through ‘pip install...

Compile issue: Tensor conv2d_bias_64_1 not in outputs for op avg_pool2d_53

3

## summary I use AITemplate to re-construct a diffusion model which is slightly different than the one in examples, but error occurs while call compile_model(). Since it says some conv2d_bias...