Reduce the sum and square-sum, compute the statistics, and broadcast it.

3x3 convolution + BN + relu

Puts each data field into a tensor with outer dimension batch size

Parse user's devices input str to standard format. e.g. [gpu0, gpu1, ...]

Monkey-patch an existing `DataParallel` object. Add the replication callback. Useful when you have customized `DataParallel` implementation.

Randomly split a dataset into non-overlapping new datasets of given lengths ds Arguments: dataset (Dataset): Dataset to be split

Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet

Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet

Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet