radubulimac's picture
first commit
7ee183c verified
from .data_setup import data_setup
from .model import create_effnetb2_model, get_transforms
from .train_and_test import train, train_step, test_step
__all__ = ['data_setup', 'create_effnetb2_model', 'get_transforms', 'train', 'train_step', 'test_step']
__version__ = '0.1.0'