search

DeepWok / mase

Machine-Learning Accelerator System Exploration Tools
Other
105 stars 52 forks source link

Fix software train runner RunnerBasicTrain (Fix for #89) #90

Closed btarifi10 closed 3 months ago

btarifi10 commented 3 months ago

Fixes the training runner (see #89)

See output after search:

|    |   number | software_metrics                   | hardware_metrics                                 | scaled_metrics                           |
|----+----------+------------------------------------+--------------------------------------------------+------------------------------------------|
|  0 |        4 | {'loss': 1.105, 'accuracy': 0.602} | {'average_bitwidth': 4.0, 'memory_density': 8.0} | {'average_bitwidth': 0.8, 'loss': 1.105} |