mlconfig

pyproject.toml

@@ -9,14 +9,14 @@ python = "^3.9"
 torch = "^2.1.0"
 torchvision = "^0.16.0"
 torchmetrics = "^0.10.1"
-omegaconf = "^2.2.3"
 mlflow = "^2.6.0"
 tqdm = "^4.64.1"
 loguru = "^0.6.0"
+mlconfig = "^0.2.0"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.7.0"
-codecov = "^2.1.13"
+coverage = "^7.3.2"
 flake8 = "^6.0.0"
 isort = "^5.12.0"
 pytest = "^7.3.1"

template/cli.py

@@ -1,14 +1,13 @@
 import click
-
-from .utils import instantiate
-from .utils import load_config
+from mlconfig import instantiate
+from mlconfig import load
 
 
 @click.command()
 @click.option("-c", "--config-file", type=click.STRING, default="configs/mnist.yaml")
 @click.option("-r", "--resume", type=click.STRING, default=None)
 def main(config_file, resume):
-    config = load_config(config_file)
+    config = load(config_file)
 
     job = instantiate(config.job)
     job.run(config, resume)

template/datasets/mnist.py

@@ -1,9 +1,8 @@
+from mlconfig import register
 from torch.utils import data
 from torchvision import datasets
 from torchvision import transforms
 
-from ..utils import register
-
 
 @register
 class MNISTDataLoader(data.DataLoader):

template/jobs/mnist.py

@@ -1,10 +1,10 @@
 import mlflow
 import torch
+from mlconfig import instantiate
+from mlconfig import register
 from omegaconf import OmegaConf
 
-from ..utils import instantiate
 from ..utils import manual_seed
-from ..utils import register
 from .job import Job
 
 

template/models/lenet.py

@@ -1,7 +1,6 @@
+from mlconfig import register
 from torch import nn
 
-from ..utils import register
-
 
 class ConvBNReLU(nn.Sequential):
     def __init__(self, in_channels, out_channels, kernel_size):

template/optim.py

@@ -1,7 +1,6 @@
+from mlconfig import register
 from torch import optim
 
-from .utils import register
-
 register(optim.Adam)
 
 register(optim.lr_scheduler.StepLR)

template/trainers/mnist.py

@@ -1,12 +1,12 @@
 import mlflow
 import torch
 import torch.nn.functional as F
+from mlconfig import register
 from torchmetrics import Accuracy
 from torchmetrics import MeanMetric
 from tqdm import tqdm
 from tqdm import trange
 
-from ..utils import register
 from .trainer import Trainer
 
 

template/utils/__init__.py

@@ -1,6 +1,3 @@
-from .conf import instantiate
-from .conf import load_config
-from .conf import register
 from .utils import load_json
 from .utils import load_yaml
 from .utils import manual_seed