7eb23e3c1d
* Initial commit * Initial commit * Cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix precommit errors * Remove TF builds from CI * export last.pt * Created using Colaboratory * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import sys
|
|
from pathlib import Path
|
|
|
|
import wandb
|
|
|
|
FILE = Path(__file__).resolve()
|
|
ROOT = FILE.parents[3] # root directory
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.append(str(ROOT)) # add ROOT to PATH
|
|
|
|
from train import parse_opt, train
|
|
from utils.callbacks import Callbacks
|
|
from utils.general import increment_path
|
|
from utils.torch_utils import select_device
|
|
|
|
|
|
def sweep():
|
|
wandb.init()
|
|
# Get hyp dict from sweep agent
|
|
hyp_dict = vars(wandb.config).get("_items")
|
|
|
|
# Workaround: get necessary opt args
|
|
opt = parse_opt(known=True)
|
|
opt.batch_size = hyp_dict.get("batch_size")
|
|
opt.save_dir = str(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok or opt.evolve))
|
|
opt.epochs = hyp_dict.get("epochs")
|
|
opt.nosave = True
|
|
opt.data = hyp_dict.get("data")
|
|
opt.weights = str(opt.weights)
|
|
opt.cfg = str(opt.cfg)
|
|
opt.data = str(opt.data)
|
|
opt.hyp = str(opt.hyp)
|
|
opt.project = str(opt.project)
|
|
device = select_device(opt.device, batch_size=opt.batch_size)
|
|
|
|
# train
|
|
train(hyp_dict, opt, device, callbacks=Callbacks())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sweep()
|