'relu_'+i, nn.ReLU() ) if layer_info['activation']['val'] == 'sigmoid': self.model.add_module( 'sigm_'+i, nn.Sigmoid() ) if layer_info['activation']['val'] == 'elu': self.model.add_module( 'elu_'+i, nn.ELU() ) previous_units = layer_info['nb_units']['val'] self.model.add_module( 'classification_layer', nn.Linear(previous_units, 10) ) self.model.add_module('sofmax', nn.LogSoftmax()) self.model.cpu() if build_info['optimizer']['val'] == 'adam': optimizer = optim.Adam(self.model.parameters(), lr=build_info['weight_decay']['val'], weight_decay=build_info['weight_decay']['val']) elif build_info['optimizer']['val'] == 'adadelta': optimizer = optim.Adadelta(self.model.parameters(), lr=build_info['weight_decay']['val'], weight_decay=build_info['weight_decay']['val']) elif build_info['optimizer']['val'] == 'rmsprop': optimizer = optim.RMSprop(self.model.parameters(), lr=build_info['weight_decay']['val'], weight_decay=build_info['weight_decay']['val']) else: optimizer = optim.SGD(self.model.parameters(),
推荐阅读
百度前首席科学家吴恩达:AI短期内不会超越人类
吴恩达认为,根本收入筹划的目标不该该是让人们持续给打车公司开车,他欲望根本收入筹划可以或许资世人们进修更多的器械,赞助他们找到有意义的工作。
腾讯科技讯 据外媒报道,固然比来有很>>>详细阅读
本文标题:如何用自动机器学习实现神经网络进化
地址:http://www.17bianji.com/lsqh/36435.html
1/2 1