(to_be_distributed)
| 76 | |
| 77 | |
| 78 | def init_data_loaders(to_be_distributed): |
| 79 | # Prepare dataset |
| 80 | train_loader = prepare_dataloader( |
| 81 | MyData(datasets=config.training_set, image_size=config.size, is_train=True), |
| 82 | config.batch_size, to_be_distributed=to_be_distributed, is_train=True |
| 83 | ) |
| 84 | print(len(train_loader), "batches of train dataloader {} have been created.".format(config.training_set)) |
| 85 | test_loaders = {} |
| 86 | for testset in args.testsets: |
| 87 | _data_loader_test = prepare_dataloader( |
| 88 | MyData(datasets=testset, image_size=config.size, is_train=False), |
| 89 | config.batch_size_valid, is_train=False |
| 90 | ) |
| 91 | print(len(_data_loader_test), "batches of valid dataloader {} have been created.".format(testset)) |
| 92 | test_loaders[testset] = _data_loader_test |
| 93 | return train_loader, test_loaders |
| 94 | |
| 95 | |
| 96 | def init_models_optimizers(epochs, to_be_distributed): |
no test coverage detected