pub struct Vjepa2Runner { /* private fields */ }Expand description
V-JEPA2 runner — encoder (+ optional predictor / pooler).
Implementations§
Source§impl Vjepa2Runner
impl Vjepa2Runner
pub fn builder() -> Vjepa2RunnerBuilder
pub fn config(&self) -> &Vjepa2Config
pub fn device(&self) -> Device
pub fn has_predictor(&self) -> bool
pub fn has_pooler(&self) -> bool
Sourcepub fn encode_video(
&mut self,
video_ncthw: &[f32],
) -> Result<Vjepa2Output, Error>
pub fn encode_video( &mut self, video_ncthw: &[f32], ) -> Result<Vjepa2Output, Error>
Encode a pre-normalized video tensor [C, T, H, W] (NCTHW f32).
Sourcepub fn encode_video_hwc(&mut self, frames: &[u8]) -> Result<Vjepa2Output, Error>
pub fn encode_video_hwc(&mut self, frames: &[u8]) -> Result<Vjepa2Output, Error>
Convenience: u8 HWC frames [num_frames, crop, crop, 3] → encode.
Sourcepub fn predict(
&mut self,
enc: &Vjepa2Output,
masks: &Vjepa2Masks,
) -> Result<Vjepa2PredictOutput, Error>
pub fn predict( &mut self, enc: &Vjepa2Output, masks: &Vjepa2Masks, ) -> Result<Vjepa2PredictOutput, Error>
Run the JEPA predictor on encoder outputs with context/target masks.
Sourcepub fn pool(&self, enc: &Vjepa2Output) -> Result<Vjepa2PoolOutput, Error>
pub fn pool(&self, enc: &Vjepa2Output) -> Result<Vjepa2PoolOutput, Error>
Attentive pooler (+ classifier when present) on encoder tokens.
Auto Trait Implementations§
impl !RefUnwindSafe for Vjepa2Runner
impl !Sync for Vjepa2Runner
impl !UnwindSafe for Vjepa2Runner
impl Freeze for Vjepa2Runner
impl Send for Vjepa2Runner
impl Unpin for Vjepa2Runner
impl UnsafeUnpin for Vjepa2Runner
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> 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 more