Refactor pl_bolts.models.self_supervised.amdim
#479
Merged
pl_bolts.models.self_supervised.amdim
#479