training loop speedup

This commit is contained in:
Joseph Hopfmüller
2024-11-20 11:29:18 +01:00
parent 1622c38582
commit cdca5de473
11 changed files with 1026 additions and 151 deletions

View File

@@ -1,2 +1,17 @@
from .datasets import FiberRegenerationDataset # noqa: F401
from .plot import eye # noqa: F401
from . import datasets # noqa: F401
# from .datasets import FiberRegenerationDataset # noqa: F401
# from .datasets import load_data # noqa: F401
from . import plot # noqa: F401
# from .plot import eye # noqa: F401
from . import optuna_helpers # noqa: F401
# from .optuna_helpers import optional_suggest_categorical # noqa: F401
# from .optuna_helpers import optional_suggest_float # noqa: F401
# from .optuna_helpers import optional_suggest_int # noqa: F401
from . import complexNN # noqa: F401
# from .complexNN import UnitaryLayer # noqa: F401
# from .complexNN import complex_mse_loss # noqa: F401
# from .complexNN import complex_sse_loss # noqa: F401