Projects Security Insights Additional navigation options Files main .gitignore README.md config.py dataset.py install-cn-qinglong.ps1 main.py optimizer.py requirements.txt train.ps1 train.py Breadcrumbs florence2-ft-advanced / Latest commit
total_size = len(dataset.data) train_size = int(args.train_split * total_size) val_size = total_size - train_size train_dataset, val_dataset = random_split(dataset, [train_size, val_size]) train_dataset.dataset.split = 'train' val_dataset.dataset.split = 'validation' train_dataset....