Medical-Transformer icon indicating copy to clipboard operation
Medical-Transformer copied to clipboard

Published 20 hours ago • verified publisher icon jeya-maria-jose

Softmax fix

Open rw404 opened this issue 1 year ago • 1 comments

  • Softmax layer in eval mode added;
  • model.train(), model.eval() in train.py placed for better performance

Fixes from issue #51

rw404 avatar Feb 21 '23 19:02 rw404