Commit 1cb3a41e authored Mar 13, 2023 by narugo1992

dev(narugo): add train script

parent c5f76e3f

.gitignore

+2 −1

Original line number	Diff line number	Diff line
		@@ -1217,3 +1217,4 @@ fabric.properties
		!/zoo/testfile/*/
		/test/testfile/dataset
		/zoo/testfile/dataset
		/.train
		No newline at end of file

requirements-zoo.txt

+2 −0

Original line number	Diff line number	Diff line
		@@ -8,3 +8,5 @@ onnxoptimizer
		onnxsim
		onnxruntime
		click
		di-toolkit
		tensorboard

requirements.txt

+0 −1

Original line number	Diff line number	Diff line
		hbutils>=0.8.2
		pillow
		numpy
		opencv-python
		scikit-learn
		huggingface_hub
		tqdm

zoo/base.py

0 → 100644

+5 −0

Original line number	Diff line number	Diff line
		from ditk import logging

		_TRAIN_DIR = '.train'

		logging.try_init_root(logging.INFO)

zoo/monochrome/init.py

+1 −0

Original line number	Diff line number	Diff line
		from .train_ import train