/SemiImplicitDNNs

Pytorch implementation of 'Semi-Implicit Methods for Deep Neural Networks'

Primary LanguagePython

SemiImplicitDNNs

Pytorch implementation of IMEXnet - A Forward Stable Deep Neural Network demonstrated on the synthetic Qtips dataset.

Use

To run IMEXnet python synth.py

To run the ResNet baseline python synth.py --net_type resnet

After running both models, you can compare the results with gen_val_plots.py and gen_loss_plots.py.