fixed dict name error for grabbing the dropout rates