dqn model data fix
This commit is contained in:
@ -1167,7 +1167,7 @@ class DQNAgent:
|
||||
|
||||
# Handle empty dictionary case
|
||||
if not state:
|
||||
logger.error("No numerical values found in state dict, using default state")
|
||||
logger.error("Empty state dictionary received, using default state")
|
||||
expected_size = getattr(self, 'state_size', 403)
|
||||
if isinstance(expected_size, tuple):
|
||||
expected_size = np.prod(expected_size)
|
||||
|
Reference in New Issue
Block a user