jrfackler

Hello @MohitBurkule , Thank you very much for your reply. Yes, you found my mistake in the example of not passing the arch_config to the Regressor. Doing that adds the...

Thank you again @MohitBurkule Your solution of adding the dropout layer afterwards works for the standard case. But, for some reason when using a cu

Your solution worked great! Thank you @MohitBurkule