2017-02-13 4 views
1

Когда я определил модель, как это:Как получить форму/размер слоев?

def create_basic_model_terse(input, out_dims): 

    with default_options(activation=relu): 
     model = Sequential([ 
      LayerStack(3, lambda i: [ 
       Convolution((5,5), [32,32,64][i], init=glorot_uniform(), pad=True), 
       MaxPooling((3,3), strides=(2,2)) 
      ]), 
      Dense(64, init=glorot_uniform()), 
      Dense(out_dims, init=glorot_uniform(), activation=None) 
     ]) 

    return model(input) 

Как я могу получить какую-то информацию о каждом слое в сети, как выходной формы/размеры?

ответ

1

Вы можете ознакомиться с учебниками CNTK 202. Существуют и другие учебные пособия, такие как CNTK 105, которые также показывают, как получить разные атрибуты моделей.

For a model 
 
def create_model(): 
 
with default_options(initial_state=0.1): 
 
    return Sequential([ 
 
     Embedding(emb_dim), 
 
     Recurrence(LSTM(hidden_dim), go_backwards=False), 
 
     Dense(num_labels) 
 
    ]) 
 

 

 

 
model = create_model() 
 
print(len(model.layers)) 
 
print(model.layers[0].E.shape) 
 
print(model.layers[2].b.value)

+0

Спасибо, я также получил TensorBoard к wotk с CNTK, так что это вариант тоже. – OlavT

Смежные вопросы