jittor框架

 

 

sudo apt install python-dev
sudo pip install https://github.com/Jittor/jittor.git
python3 -m jittor.test.test_example

http://cg.cs.tsinghua.edu.cn/jittor/

 

ssd网络有预训练:

https://github.com/Jittor/ssd-jittor

转python:

   images = jt.array(images)  # (batch_size (N), 3, 300, 300)
        boxes = [jt.array(b) for b in boxes]
        labels = [jt.array(l) for l in labels]
        # jt.sync_all(True)
        # print("1: ", time.time() - start)

        # start = time.time()
        # jt.sync_all(True)
        predicted_locs, predicted_scores = model(images)

 

旷世天元框架

Brain++

你可能感兴趣的:(深度学习基础,cuda)