Skip to content

A toolbox to analyze the robustness of given neural networks and a set of robust low-rank training method

License

Notifications You must be signed in to change notification settings

ScSteffen/RobustDLRT

Repository files navigation

Robust and efficient low-rank compression and transfer learning models for geospatial applications.

Publications

Installation and package management

  1. Clone the Github repository:

    git clone https://github.com/ScSteffen/RobustDLRT.git
    
  2. Create a local python environment and install the python requirements in a local virtual environment:

    python3 -m venv ./venv
    source venv/bin/activate
    pip install -r requirements.txt
    

Set up datasets

  1. put the data into .dataset/data_adversarial_rs/<testcasename>. The easiest way to get started is to run
   cd dataset
   python create_cifar10.py

Example run

sh run_example.sh

Testing Pipeline for OReole-FM

To run experiments on OReole-FM MR models, the transformers need to be converted from the timm format to transformers format.

Run convert_timm_to_hf_vit.py with the appropriate command line arguments.

Authors (alphabetically ordered)

Main contributors

  • Schnake, Stefan
  • Schotthoefer, Steffen
  • Yang, Lexie H.

Student contributors

  • Snyder, Thomas
  • Park, Hannah

About

A toolbox to analyze the robustness of given neural networks and a set of robust low-rank training method

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published