popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

COB integration and refactoring

Browse Source
This commit is contained in:
Dobromir Popov
2025-06-25 02:48:00 +03:00
parent afefcea308
commit e57c6df7e1
6 changed files with 564 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
NN/models/__init__.py
View File

@ -15,5 +15,6 @@ from NN.models.transformer_model_pytorch import (
TransformerModelPyTorch as TransformerModel,
MixtureOfExpertsModelPyTorch as MixtureOfExpertsModel
)
from NN.models.cob_rl_model import MassiveRLNetwork, COBRLModelInterface
__all__ = ['CNNModel', 'TransformerModel', 'MixtureOfExpertsModel']
__all__ = ['CNNModel', 'TransformerModel', 'MixtureOfExpertsModel', 'MassiveRLNetwork', 'COBRLModelInterface']