blob: 1b5983a1bfa5077edb8115e7d9b6c1aa6ee31e8b (
plain)
1
2
3
4
5
6
7
8
9
10
11
|
import learning.nets.fc_2layers_1024units as fc_2layers_1024units
def build_net(net_name, input_tfs, reuse=False):
net = None
if (net_name == fc_2layers_1024units.NAME):
net = fc_2layers_1024units.build_net(input_tfs, reuse)
else:
assert False, 'Unsupported net: ' + net_name
return net
|