BioroboticsLab / IBA

Information Bottlenecks for Attribution
MIT License
74 stars 9 forks source link

add tensorflow implementation & update api #3

Closed berleon closed 4 years ago

berleon commented 4 years ago

the idea is to have both tensorflow and pytorch implementation in this repo. a possible api could be:

iba.tensorflow.*
iba.pytorch.* 
iba.common.*

where common would host all shared things (e.g. visualization).