chore: update dataset_loader.cpp (#6483)

This commit is contained in:
Ikko Eltociear Ashimine 2024-06-17 11:33:43 +09:00 коммит произвёл GitHub
Родитель b96d02404a
Коммит d56a7a3c36
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -274,7 +274,7 @@ Dataset* DatasetLoader::LoadFromFile(const char* filename, int rank, int num_mac
dataset.reset(LoadFromBinFile(filename, bin_filename.c_str(), rank, num_machines, &num_global_data, &used_data_indices));
// checks whether there's a initial score file when loaded from binary data files
// the intial score file should with suffix ".bin.init"
// the initial score file should with suffix ".bin.init"
dataset->metadata_.LoadInitialScore(bin_filename);
dataset->device_type_ = config_.device_type;
@ -344,7 +344,7 @@ Dataset* DatasetLoader::LoadFromFileAlignWithOtherDataset(const char* filename,
// load data from binary file
dataset.reset(LoadFromBinFile(filename, bin_filename.c_str(), 0, 1, &num_global_data, &used_data_indices));
// checks whether there's a initial score file when loaded from binary data files
// the intial score file should with suffix ".bin.init"
// the initial score file should with suffix ".bin.init"
dataset->metadata_.LoadInitialScore(bin_filename);
}
// not need to check validation data