export_to_tensorflow.py halts without any error

alexlee-gk / lpips-tensorflow

Tensorflow port for the Learned Perceptual Image Patch Similarity (LPIPS) metric.

BSD 2-Clause "Simplified" License

130 stars 30 forks source link

export_to_tensorflow.py halts without any error #11

Open harbarex opened 4 years ago

harbarex commented 4 years ago

When the line tf_rep = prepare(model, device='CPU') is reached in export_to_tensorflow.py, the script halts without any warning or message. And subsequently, no model is saved in the .lpips folder. What can be the reason for this?

I am using tensorflow 2.1.1