Обновление tf.contrib.slim вручную до tf 2.0

Question

У меня проблема с моим кодом на Python, который использует функции tf.contrib.slim и больше не работает после обновления до tenorflow до 2.0.

Как я могу обновить следующее до tf 2.0:

import tensorflow.contrib.slim as slim
import tensorflow.contrib.slim.nets

# ...

net = slim.conv2d(
    inp, 
    dim,
    [3, 3],
    rate=1,
    normalizer_fn=slim.layer_norm,
    activation_fn=lrelu,
    scope='g_' + str(width) + '_conv1') 

Спасибо.