def set_network(self, net_name: str = 'fmnist_LeNet_one_class'):
"""
Set the network structure for the model.
The key here is to initialize <self.net>.
"""
self.net_name = net_name
# Note that <build_network> is different from <build_autoencoder>
# Yet they share the same parameter
self.net = build_network(net_name)
def set_network(self, net_name):
self.net_name = net_name
self.net = build_network(net_name)
def set_network(self, net_name):
"""
The key here is to initialize <self.net>.
"""
self.net_name = net_name
self.net = build_network(net_name)