在用keras中的问题

kidcad

在强化学习中，使用了较高版本的tensorflow

1. def create_actor_network(self, state_size,action_dim):
   
2.         print("Now we build the model")
   
3.         S = Input(shape=[state_size])   
   
4.         h0 = Dense(HIDDEN1_UNITS, activation='relu')(S)
   
5.         h1 = Dense(HIDDEN2_UNITS, activation='relu')(h0)
   
6.         Steering = Dense(1,activation='tanh',init=lambda shape, name: normal(shape, scale=1e-4, name=name))(h1)
   
7.         Acceleration = Dense(1,activation='sigmoid',init=lambda shape, name: normal(shape, scale=1e-4, name=name))(h1)
   
8.         Brake = Dense(1,activation='sigmoid',init=lambda shape, name: normal(shape, scale=1e-4, name=name))(h1)
   
9.         V = merge([Steering,Acceleration,Brake],mode='concat')
   
10.         model = Model(input=S,output=V)
    
11.         return model, model.trainable_weights, S

复制代码

结果报错

1. File "/home/gym_torcs-master/ddpg.py", line 52, in playGame
   
2.     actor = ActorNetwork(sess, state_dim, action_dim, BATCH_SIZE, TAU, LRA)
   
3.   File "/home/gym_torcs-master/ActorNetwork.py", line 25, in __init__
   
4.     self.model , self.weights, self.state = self.create_actor_network(state_size, action_size)   
   
5.   File "/home/gym_torcs-master/ActorNetwork.py", line 51, in create_actor_network
   
6.     Steering = Dense(1,activation='tanh',init=lambda shape, name: normal(shape, scale=1e-4, name=name))(h1)
   
7.   File "/home/anaconda3/lib/python3.6/site-packages/keras/engine/base_layer.py", line 432, in __call__
   
8.     self.build(input_shapes[0])
   
9.   File "/home/anaconda3/lib/python3.6/site-packages/keras/layers/core.py", line 872, in build
   
10.     constraint=self.kernel_constraint)
    
11.   File "/home/anaconda3/lib/python3.6/site-packages/keras/legacy/interfaces.py", line 91, in wrapper
    
12.     return func(*args, **kwargs)
    
13.   File "/home/anaconda3/lib/python3.6/site-packages/keras/engine/base_layer.py", line 249, in add_weight
    
14.     weight = K.variable(initializer(shape),
    
15. TypeError: <lambda>() missing 1 required positional argument: 'name'

复制代码

塔利班

代码不完整，只能看提示，lambda表达式缺参数name

kidcad

最后还是降了keres版本，然后就不报错了，不知道为什么。。