pub struct PackNet { /* private fields */ }Expand description
Progressive Networks (PackNet) optimizer.
Implementations§
Source§impl PackNet
impl PackNet
Sourcepub fn new(
config: PackNetConfig,
initial_parameters: Vec<Tensor>,
) -> Result<Self>
pub fn new( config: PackNetConfig, initial_parameters: Vec<Tensor>, ) -> Result<Self>
Create a new PackNet optimizer.
Sourcepub fn allocate_task(&mut self, task_id: usize) -> Result<()>
pub fn allocate_task(&mut self, task_id: usize) -> Result<()>
Allocate parameters for a new task.
Sourcepub fn step(&mut self, gradients: &[Tensor]) -> Result<()>
pub fn step(&mut self, gradients: &[Tensor]) -> Result<()>
Perform optimization step with parameter masking.
Sourcepub fn get_parameters(&self) -> &[Tensor]
pub fn get_parameters(&self) -> &[Tensor]
Get current parameters.
Sourcepub fn get_available_capacity(&self) -> &[f32]
pub fn get_available_capacity(&self) -> &[f32]
Get available capacity for new tasks.
Auto Trait Implementations§
impl Freeze for PackNet
impl RefUnwindSafe for PackNet
impl Send for PackNet
impl Sync for PackNet
impl Unpin for PackNet
impl UnwindSafe for PackNet
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
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 more