typewriter
typewriter copied to clipboard
Using custom tf.keras model
Is it possible to use a custom tf.keras model I made as the network or middleware in e.g. the Rainbow agent? My network takes images as input and has as many output neurons as there are actions. I haven't found how to use a custom network in the documentation or tutorial.