From 336594dfa4eb286e33f785edaf8a87800a23a2ad Mon Sep 17 00:00:00 2001 From: Eason <30045503+Eason0729@users.noreply.github.com> Date: Wed, 12 Jun 2024 23:24:22 +0800 Subject: [PATCH] remove check every feature --- tank-rust/src/dqn/feature.rs | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) 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();