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

A single linear layer which does a 1x1 conv

Dataset for CP-VTON.

Test Dataset for CP-VTON.

takes 2 distances, puts through fc layers, spits out value between [0,1] (if use_sigmoid is True)

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

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

conditional batchnorm

Interpolation module.

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