Merge branch 'main' into unit_testing

# Conflicts:
#	marl_factory_grid/modules/doors/groups.py
#	marl_factory_grid/utils/states.py
This commit is contained in:
Chanumask
2023-11-23 12:58:12 +01:00
63 changed files with 1477 additions and 330 deletions

View File

@ -1,7 +1,6 @@
from pathlib import Path
import numpy as np
import torch
import yaml
@ -74,6 +73,7 @@ class Checkpointer(object):
def save_experiment(self, name: str, model):
cpt_path = self.path / f'checkpoint_{self.__current_checkpoint}'
cpt_path.mkdir(exist_ok=True, parents=True)
import torch
torch.save(model.state_dict(), cpt_path / f'{name}.pt')
def step(self, to_save):