def __init__(self, state_dim = STATE_DIM): super(Q, self).__init__( l1=F.Linear(state_dim, 50), l2=F.Linear(50, 50), q_value=F.Linear(50, 5) )