MATLAB Answers

Can I use NN built with Fitnet for Reinforcement Learning toolbox with DQN agent?

1 view (last 30 days)
Hello ,
Can I reuse the neural network built with Fitnet fuction (Function fitting neural network), inside Reinforcement Learning toolbox with DQN agent?
The neural network does not look like layers as in the Deep Network Designer. I tried using them but not able to provide the 'ImageInputLayer' to the fitnet Network.
critic = rlQValueRepresentation(net1,observationInfo,actionInfo,'Observation',{net1.inputs{1}.name});
I tried above also but still not working.
Anybody tried this method or any suggestion for a work around?


Sign in to comment.

Accepted Answer

Emmanouil Tzorakoleftherakis
Hi Abhay,
Reinforcement Learning Toolbox currently supports the layers supported by Deep Learning Toolbox only. You could try converting your network using Deep Learning Toolbox layers after extracting the trained parameters.

More Answers (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!