fixes
This commit is contained in:
@ -229,8 +229,8 @@ class COBRLModelInterface(ModelInterface):
|
||||
Interface for the COB RL model that handles model management, training, and inference
|
||||
"""
|
||||
|
||||
def __init__(self, model_checkpoint_dir: str = "models/realtime_rl_cob", device: str = None):
|
||||
super().__init__(name="cob_rl_model") # Initialize ModelInterface with a name
|
||||
def __init__(self, model_checkpoint_dir: str = "models/realtime_rl_cob", device: str = None, name=None, **kwargs):
|
||||
super().__init__(name=name) # Initialize ModelInterface with a name
|
||||
self.model_checkpoint_dir = model_checkpoint_dir
|
||||
self.device = torch.device(device if device else ('cuda' if torch.cuda.is_available() else 'cpu'))
|
||||
|
||||
|
Reference in New Issue
Block a user