>> import tools

>>> tools.float_performance()

==> float type : float64

Dataset is loaded...train 60000 and test 10000

Epoch 1:4131/10000,Cost(on train):1.976069012858489,round time(s):27.29

Epoch 2:6981/10000,Cost(on train):1.2734770169398337,round time(s):27.15

Epoch 3:8059/10000,Cost(on train):0.863751833232319,round time(s):26.75

check weight : float64

check bias : float64

check data : float64

==> float type : float32

Dataset is loaded...train 60000 and test 10000

Epoch 1:5542/10000,Cost(on train):1.9846122093061607,round time(s):20.16

Epoch 2:7300/10000,Cost(on train):1.2205352244853973,round time(s):20.58

Epoch 3:8176/10000,Cost(on train):0.8342689389915516,round time(s):20.24

check weight : float32

check bias : float32

check data : float32

==> float type : dataset in float16, nn in float32

Dataset is loaded...train 60000 and test 10000

Epoch 1:4480/10000,Cost(on train):2.0938571048120656,round time(s):22.39

Epoch 2:6940/10000,Cost(on train):1.3261136438975731,round time(s):22.14

Epoch 3:8118/10000,Cost(on train):0.8644946610357612,round time(s):22.46

check weight : float32

check bias : float32

check data : float16

==> float type : float16

Dataset is loaded...train 60000 and test 10000

Epoch 1:3810/10000,Cost(on train):2.0293811360677085,round time(s):133.39

Epoch 2:7130/10000,Cost(on train):1.3160956583658854,round time(s):133.8

Epoch 3:8092/10000,Cost(on train):0.9138703043619791,round time(s):135.69

check weight : float16

check bias : float16

check data : float16

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐