Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up

BuilT - Build a Trainer of deep neural networks

License

NotificationsYou must be signed in to change notification settings

UoA-CARES/BuilT

Repository files navigation

Build Statuscodecov

BuilT(Build a Trainer)

Easily build a trainer for your Depp Neural Network model and experiment as many as you want to find optimal combination of components(model, optimizer, scheduler) and hyper-parameters in a well-organized manner.

  • No more boilerplate code to train and evaluate your DNN model. just focus on your model.
  • Simply swap your dataset, model, optimizer and scheduler in the configuration file to find optimal combination. Your code doesn't need to be changed!!!.
  • Support Cross Validation, OOF(Out of Fold) Prediction
  • Support WandB(https://wandb.ai/) or tensorboard logging.
  • Support checkpoint management(Save and load a model. Resume the previous training)
  • BuilT easily integrates with Kaggle(https://www.kaggle.com/) notebook. (todo: add notebook link)

Installation

Please follow the instruction below to install BuilT.

Installation of BuilT package from the source code

git clone https://github.com/UoA-CARES/BuilT.gitcd BuilTpython setup.py install

Installation of BuilT package using pip

BuilT can be installed using pip(https://pypi.org/project/BuilT/).

pip install built

Usage

Configuration

Builder

Trainer

Dataset

Model

Loss

Optimizer

Scheduler

Logger

Metric

Inference

Ensemble

Examples

MNIST hand-written image classification

(todo)

Sentiment Classification

(todo)

Developer Guide

(todo)

conda create -n conda_BuilT python=3.7conda activate conda_BuilTpip install -r requirements.txt

Reference

https://packaging.python.org/tutorials/packaging-projects/


[8]ページ先頭

©2009-2025 Movatter.jp