use guiding agent

2024-07-22 19:10:40 +08:00
parent 1f217770b7
commit 2d55c655c3
3 changed files with 164 additions and 2 deletions
--- a/reinforcement/model.py
+++ b/reinforcement/model.py
@ -11,7 +11,7 @@ from torch import tensor, double, optim
 from torch.nn.functional import relu, mse_loss
 import torch

-kProductionMode=False
+kProductionMode=True
 class DeepQNetwork(Module):
    """
    A model that uses a Deep Q-value Network (DQN) to approximate Q(s,a) as part