tf.contrib.layers.convolution2d를 사용할 때 매개 변수를 공유합니다.

Question

저는 두 세트의 이미지를 병렬로 처리하는 신경망을 구축 중입니다. 두 개의 열이 매개 변수를 공유하도록합니다. 이것이 제가하는 것입니다.

with tf.variable_scope(layer_name) as s: h1 = tf.contrib.layers.convolution2d(inputs = x1, num_outputs = 10, kernel_size = [3, 3], stride = [1, 1], padding = 'VALID', scope = s) h2 = tf.contrib.layers.convolution2d(inputs = x2, num_outputs = 10, kernel_size = [3, 3], stride = [1, 1], padding = 'VALID', reuse = True, scope = s)

이이 작업을 수행하는 올바른 방법이 있나요? tf.contrib.layers.convolution2d 클래스를 사용할 때 올바르게 수행하는 방법의 예제를 찾을 수 없습니다. 정확하지

Answer 1

은과 같이 함수로합니다가 같은 무게를 재사용 할 것이다 다른 입력을 불렀다 지금

def function(x, reuse): 
    with tf.variable_scope(layer_name) as s: 
    output = tf.contrib.layers.convolution2d(inputs = x, num_outputs = 10, kernel_size = [3, 3], 
        stride = [1, 1], padding = 'VALID', reuse = reuse, scope = s) 
    return output 

output1 = function(image1, False) 
output2 = function(image2, True) 

합니다.