diff --git a/tank-rust/src/dqn/feature.rs b/tank-rust/src/dqn/feature.rs index ea6acca..b4a2279 100644 --- a/tank-rust/src/dqn/feature.rs +++ b/tank-rust/src/dqn/feature.rs @@ -124,7 +124,7 @@ impl<'a> Info<'a> { let angle = self.player.get_angle(); let gun_angle = self.player.get_gun_angle(); - let feature=[ + [ normalize_angle(target.angle - angle).tanh(), (wall.distance - target.distance).tanh(), (self.player.power as f32).tanh(), @@ -132,11 +132,7 @@ impl<'a> Info<'a> { (emeny.distance + 1.0).log2(), normalize_angle(emeny.angle - gun_angle).tanh(), normalize_angle(wall.angle - gun_angle).tanh(), - ]; - for f in feature.iter() { - assert!(!f.is_nan(),"feature is nan: {:?}", feature); - } - feature + ] } pub fn into_feature_tensor(&self, device: &B::Device) -> Tensor { let feature = self.into_feature();