Struct OfflineDataset 

pub struct OfflineDataset {
    pub transitions: Vec<Transition>,
}

Static offline transitions for flow-matching pretraining.

Fields

transitions: Vec<Transition>

Implementations

impl OfflineDataset

pub fn from_transitions(transitions: Vec<Transition>) -> Self

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn from_replay(buf: &ReplayBuffer) -> Self

pub fn sample_diag_batch( &self, spec: &RlSpec, indices: &[usize], rng: &mut u64, ) -> ActorTrainBatch

L_Diag batch (Eq. 3): r = t, target velocity a₁ - a₀.

pub fn sample_esd_inputs( &self, spec: &RlSpec, indices: &[usize], step: usize, rng: &mut u64, ) -> EsdInputsBatch

Pack ESD inputs: states, bridge a_r, curriculum (r,t), noise velocity v_rt.

pub fn sample_cfm_batch( &self, spec: &RlSpec, indices: &[usize], rng: &mut u64, ) -> CfmBatch

Legacy CFM batch (diagonal only).

Trait Implementations

impl Clone for OfflineDataset

fn clone(&self) -> OfflineDataset

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for OfflineDataset

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for OfflineDataset

fn default() -> OfflineDataset

Returns the “default value” for a type.