codes.train.train_fcts

Functions

create_task_list_for_surrogate(config, surr_name)	Creates a list of training tasks for a specific surrogate model based on the configuration file.
parallel_training(tasks, device_list, ...)	Execute the training tasks in parallel on multiple devices.
sequential_training(tasks, device_list, ...)	Execute the training tasks sequentially on a single device.
train_and_save_model(surr_name, mode, ...[, ...])	Train and save a model for a specific benchmark mode.
worker(task_queue, device, device_idx, ...)	Worker function to process tasks from the task queue on the given device.