GitLab

* add test for AttentionBlock, SpatialTransformer

* add context_dim, handle device

* removed dropout test

* fixes, add dropout test

* Revive Make utils

* Add datasets for training too

* make unet rl work

* uploaad files / code

* upload files

* make style correct

* finish