update save and load
This commit is contained in:
+4
-4
@@ -43,13 +43,13 @@ class ExperienceBuffer:
|
||||
self.rewards = rewards
|
||||
self.advantages = advantages
|
||||
|
||||
def save(self, filename):
|
||||
with open(filename, 'wb') as f:
|
||||
def save(self, file_path):
|
||||
with open(file_path, 'wb') as f:
|
||||
pickle.dump((self.states, self.actions, self.rewards, self.advantages), f)
|
||||
|
||||
@classmethod
|
||||
def load(cls, filename):
|
||||
with open(filename, 'rb') as f:
|
||||
def load(cls, file_path):
|
||||
with open(file_path, 'rb') as f:
|
||||
states, actions, rewards, advantages = pickle.load(f)
|
||||
return cls(
|
||||
states=states,
|
||||
|
||||
Reference in New Issue
Block a user