mjtb49 / StrongholdTrainer

MIT License
22 stars 10 forks source link

Make input batching a configurable parameter #68

Closed fsharpseven closed 3 years ago

fsharpseven commented 3 years ago

Currently, whether or not a model uses input batching (by stronghold) is determined by checking if it has "rnn" in its identifier, which is a terrible check. A true/false or something would work for this.