saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python/examples
History
Luke Yeager f7ece79949 Add fp16 and tensorcore support to resnet50_trainer 
Summary:
Use like `--dtype=float16 --enable-tensor-core`
Closes https://github.com/caffe2/caffe2/pull/1093

Differential Revision: D5634840

Pulled By: harouwu

fbshipit-source-id: 18c1e70236ba5ef8661ff55fb524caae1be19310
2017-08-17 15:16:24 -07:00
..
char_rnn.py Fixed typo 2017-06-23 14:02:40 -07:00
lmdb_create_example.py Deprecate CNNModelHelper in lmdb_create_example 2017-06-19 13:04:02 -07:00
resnet50_trainer.py Add fp16 and tensorcore support to resnet50_trainer 2017-08-17 15:16:24 -07:00