Skip to main content

DropBlock

Struct DropBlock 

Source
pub struct DropBlock {
    pub block_size: usize,
    pub drop_prob: f64,
    /* private fields */
}
Expand description

DropBlock regularization.

Drops contiguous blocks of activations in convolutional feature maps. This is more effective than standard dropout for CNNs because it forces the network to learn more distributed representations.

Fields§

§block_size: usize

Size of the block to drop (e.g., 7 for 7x7 blocks)

§drop_prob: f64

Probability that a block center will be chosen for dropping

Implementations§

Source§

impl DropBlock

Source

pub fn new(block_size: usize, drop_prob: f64) -> TrainResult<Self>

Create a new DropBlock regularizer.

§Arguments
  • block_size - Size of the block to drop (must be odd and >= 1)
  • drop_prob - Probability of dropping a block (0.0 to 1.0)
§Returns

A new DropBlock instance or an error if parameters are invalid.

§Example
use tensorlogic_train::DropBlock;

let dropblock = DropBlock::new(7, 0.1).unwrap();
Source

pub fn set_drop_prob(&mut self, drop_prob: f64) -> TrainResult<()>

Set the drop probability (useful for scheduling).

§Arguments
  • drop_prob - New drop probability (0.0 to 1.0)
Source

pub fn apply( &self, activations: &ArrayView2<'_, f64>, training: bool, rng: &mut StdRng, ) -> TrainResult<Array2<f64>>

Apply DropBlock to activations.

§Arguments
  • activations - Input activation map (height × width)
  • training - Whether in training mode (drops blocks) or inference mode (no dropping)
  • rng - Random number generator
§Returns

Activation map with blocks dropped (if training) or unchanged (if inference)

§Algorithm
  1. Sample Bernoulli mask for each position (potential block centers)
  2. Expand each selected center to a block of size block_size × block_size
  3. Zero out the selected blocks
  4. Normalize by keep_prob to maintain expected value

Trait Implementations§

Source§

impl Clone for DropBlock

Source§

fn clone(&self) -> DropBlock

Returns a duplicate of the value. Read more
1.0.0 · Source§

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

Performs copy-assignment from source. Read more
Source§

impl Debug for DropBlock

Source§

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

Formats the value using the given formatter. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

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 more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

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
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V