コード例 #1
0
def test_caffe_train(net):
    Caffe().train(net)
    assert len(net.weights()) > 0
    print(net.weights())
    assert net.iteration_of_weights(net.highest_iteration_weights()) == 1000
コード例 #2
0
def test_caffe_test(net):
    caffe = Caffe()
    caffe.train(net)
    for w in net.weights():
        assert caffe.test(net, w) > 0.96
コード例 #3
0
def test_caffe_time(net):
    deploy_timings = Caffe().time(net, iterations=1)
    train_timings = Caffe().time(net, iterations=1, use_train_model=True)
    assert type(deploy_timings['avg_forward']) == float
    assert type(train_timings['avg_backward']) == float