Python batchnorm_fc_with_inputsの例

コード例 #1

 def branch(i, img, label, mod_two):
     if mod_two:
         predicate = ((i % 2) == 0)
     else:
         predicate = ((i % 2) != 0)
     return layers.cond(predicate, lambda: simple_fc_net_with_inputs(img, label, class_num=10),
                        lambda: batchnorm_fc_with_inputs(img, label, class_num=10))

コード例 #2

 def branch(i, img, label):
     return layers.cond(
         (i % 2) == 0,
         lambda: simple_fc_net_with_inputs(img, label, class_num=10),
         lambda: batchnorm_fc_with_inputs(img, label, class_num=10))

コード例 #3

 def cond_func(i, img, label):
     predicate = ((i % 2) == 0)
     return layers.cond(
         predicate,
         lambda: simple_fc_net_with_inputs(img, label, class_num=10),
         lambda: batchnorm_fc_with_inputs(img, label, class_num=10))