This code is adapted from https://github.com/AndyShih12/mac for training any-order ARMs.
The trained AO-ARMs are then used to learn the marginal network in MaM for the MLE setting.


--------------------

## Installation
```
pip install -r requirements.txt
```

## Commands for training AO-ARMs

The current batch sizes assume a GPU with 80GB memory.

```
python image_main.py

python mol_main.py 

python lang_main.py
```