update readme

This commit is contained in:
Zhiming Hu 2025-06-03 21:11:04 +02:00
parent 35ee4b75e8
commit 249a01f342
18 changed files with 4936 additions and 0 deletions

6
model/__init__.py Normal file
View file

@ -0,0 +1,6 @@
from typing import Union
from .unet import BeatGANsUNetModel, BeatGANsUNetConfig, GCNUNetModel, GCNUNetConfig
from .unet_autoenc import BeatGANsAutoencConfig, BeatGANsAutoencModel, GCNAutoencConfig, GCNAutoencModel
Model = Union[BeatGANsUNetModel, BeatGANsAutoencModel, GCNUNetModel, GCNAutoencModel]
ModelConfig = Union[BeatGANsUNetConfig, BeatGANsAutoencConfig, GCNUNetConfig,GCNAutoencConfig]