Self Attention Layer

Crops the given PIL Image on the long edge. Args: size (sequence or int): Desired output size of the crop. If size is an int instead

A thread-safe future implementation. Used only as one-to-one pipe.

r"""Samples elements randomly over multiple epochs Arguments: data_source (Dataset): dataset to sample from num_epochs (int) : Number o

Crops the given PIL Image on the long edge with a random start point. Args: size (sequence or int): Desired output size of the crop. If size i

Pipe for master-slave communication.

An abstract `SyncMaster` object. - During the replication, as the data parallel will trigger an callback of each module, all slave devices should

A re-implementation of batch normalization, used for testing the numerical stability. Author: acgtyrant See also: https://github

Data Parallel with a replication callback. An replication callback `__data_parallel_replicate__` of each module will be invoked after being

A generic data loader where the images are arranged in this way: :: root/dogball/xxx.png root/dogball/xxy.png root/dogball/xxz.

r"""Applies Synchronized Batch Normalization over a 2d or 3d input that is seen as a mini-batch. .. math:: y = \frac{x - mean[x]}{ \

r"""Applies Batch Normalization over a 4d input that is seen as a mini-batch of 3d inputs .. math:: y = \frac{x - mean[x]}{ \sqrt{Va

r"""Applies Batch Normalization over a 5d input that is seen as a mini-batch of 4d inputs .. math:: y = \frac{x - mean[x]}{ \sqrt{Va