This repository is the official implementation of Stable Representation Learning for Neural Network.
To install requirements:
pip install -r requirements.txt
To train and evaluate the model(s) with StableReprLearning framework in the paper, run this command:
python -m entry.srl_framework.<model> <number_of_experiments: 15> <early_stop_patience: 10>
To train and evaluate the model(s) with Normal framework, run this command:
python -m entry.normal_framework.<model> <number_of_experiments: 15> <early_stop_patience: 10>