Cards tested (so far):
| Model | Released |
|---|---|
| Radeon RX 550 | April 2017 |
| Radeon RX 560 | April 2017 |
| Radeon HD 7800 Series | March 2012 |
| # Our drawing graph functions. We rely / have borrowed from the following | |
| # python libraries: | |
| # https://github.com/szagoruyko/pytorchviz/blob/master/torchviz/dot.py | |
| # https://github.com/willmcgugan/rich | |
| # https://graphviz.readthedocs.io/en/stable/ | |
| def draw_graph(start, watch=[]): | |
| from graphviz import Digraph |
Cards tested (so far):
| Model | Released |
|---|---|
| Radeon RX 550 | April 2017 |
| Radeon RX 560 | April 2017 |
| Radeon HD 7800 Series | March 2012 |
| def dot_product(x, kernel): | |
| """ | |
| Wrapper for dot product operation, in order to be compatible with both | |
| Theano and Tensorflow | |
| Args: | |
| x (): input | |
| kernel (): weights | |
| Returns: | |
| """ | |
| if K.backend() == 'tensorflow': |