torsh-backend 0.1.2

Backend abstraction layer for ToRSh
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

//! # TrainingScheduler - Trait Implementations
//!
//! This module contains trait implementations for `TrainingScheduler`.
//!
//! ## Implemented Traits
//!
//! - `Default`
//!
//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)

use super::types::{ResourceLimits, ScheduleType, TrainingScheduler};

impl Default for TrainingScheduler {
    fn default() -> Self {
        Self {
            schedule_type: ScheduleType::Prioritized,
            priority_queue: Vec::new(),
            resource_limits: ResourceLimits {
                max_memory: 8 * 1024 * 1024 * 1024,
                max_cpu_cores: 8,
                max_gpu_memory: 4 * 1024 * 1024 * 1024,
                max_concurrent_tasks: 4,
            },
            concurrent_training: true,
        }
    }
}