15 lines
418 B
Python
15 lines
418 B
Python
"""DC-QMIX training wrapper."""
|
|
from agents.dcqmix_agent import DCQMIXAgent
|
|
from training.train_value_based import train_sumo_value_based
|
|
|
|
|
|
def train_sumo_dcqmix(log_dir=None, checkpoint_dir=None, run_timestamp=None):
|
|
return train_sumo_value_based(
|
|
"dcqmix",
|
|
"DC-QMIX",
|
|
DCQMIXAgent,
|
|
log_dir=log_dir,
|
|
checkpoint_dir=checkpoint_dir,
|
|
run_timestamp=run_timestamp,
|
|
)
|