pytorch interface

Here is a set of examples describing how to use WDRO for custom models via pytorch modules.

Simple Neural Network

Simple Neural Network

Polynomial regression

Polynomial regression