3 ms·

torch.nn offers a very similar interface to Keras (e.g. see Alexnet definition at https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py#L1

by apaszke 10y ago

torch.nn offers a very similar interface to Keras (e.g. see Alexnet definition at https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py#L13 https://github.com/pytorch/vision/blob/master/torchvision/mo...).