mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-01-10 00:59:34 +01:00
Fix import of DEFAULT_CHECKPOINT_DIR
This commit is contained in:
parent
160cb6ae30
commit
24dcc47e73
@ -1,3 +1,4 @@
|
|||||||
|
from batdetect2.train.checkpoints import DEFAULT_CHECKPOINT_DIR
|
||||||
from batdetect2.train.config import (
|
from batdetect2.train.config import (
|
||||||
TrainingConfig,
|
TrainingConfig,
|
||||||
load_train_config,
|
load_train_config,
|
||||||
@ -6,7 +7,7 @@ from batdetect2.train.lightning import (
|
|||||||
TrainingModule,
|
TrainingModule,
|
||||||
load_model_from_checkpoint,
|
load_model_from_checkpoint,
|
||||||
)
|
)
|
||||||
from batdetect2.train.train import DEFAULT_CHECKPOINT_DIR, build_trainer, train
|
from batdetect2.train.train import build_trainer, train
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"DEFAULT_CHECKPOINT_DIR",
|
"DEFAULT_CHECKPOINT_DIR",
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user