Spaces:
Runtime error
Runtime error
File size: 305 Bytes
64ec292 | 1 2 3 4 5 6 7 8 9 10 | from utils.dataset import prepare_data
from utils.settings import get_model_from_config, parse_args_train
args = parse_args_train(None)
_, config = get_model_from_config(args.model_type, args.config_path)
batch_size = config.training.batch_size * args.device_ids
prepare_data(config, args, batch_size)
|