pub struct TrainConfig {
pub base_dir: PathBuf,
pub tokenizer: PathBuf,
pub epochs: usize,
pub lr: f64,
pub batch: usize,
pub max_len: usize,
pub lambda_b: f64,
pub seed: u64,
}Fields§
§base_dir: PathBufHF snapshot dir holding config.json + model.safetensors (e.g. bert-tiny).
tokenizer: PathBufa tokenizer.json compatible with the base’s vocab
epochs: usize§lr: f64§batch: usize§max_len: usize§lambda_b: f64weight on Head B’s loss relative to Head A
seed: u64Trait Implementations§
Source§impl Clone for TrainConfig
impl Clone for TrainConfig
Source§impl Debug for TrainConfig
impl Debug for TrainConfig
Auto Trait Implementations§
impl Freeze for TrainConfig
impl RefUnwindSafe for TrainConfig
impl Send for TrainConfig
impl Sync for TrainConfig
impl Unpin for TrainConfig
impl UnsafeUnpin for TrainConfig
impl UnwindSafe for TrainConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
Source§fn in_current_span(self) -> Instrumented<Self> ⓘ
fn in_current_span(self) -> Instrumented<Self> ⓘ
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreCreates a shared type from an unshared type.