Add evaluation slot attention + visualisation script
Showing
- eval_sa.py 55 additions, 0 deletionseval_sa.py
- osrt/layers.py 36 additions, 37 deletionsosrt/layers.py
- osrt/model.py 13 additions, 13 deletionsosrt/model.py
- osrt/utils/visualize.py 27 additions, 1 deletionosrt/utils/visualize.py
- runs/clevr3d/slot_att/config.yaml 1 addition, 1 deletionruns/clevr3d/slot_att/config.yaml
- train_sa.py 3 additions, 0 deletionstrain_sa.py
Loading
Please register or sign in to comment