From c9c23488106c024a1de08968649577c4a3c014ac Mon Sep 17 00:00:00 2001
From: sandyhouse <lilong12@baidu.com>
Date: Mon, 17 Feb 2020 16:28:17 +0800
Subject: [PATCH] add fp16

---
 demo/dali_reader.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/demo/dali_reader.py b/demo/dali_reader.py
index 3efde21..00cea87 100644
--- a/demo/dali_reader.py
+++ b/demo/dali_reader.py
@@ -38,6 +38,7 @@ def main():
     ins.set_dataset_dir(args.data_dir)
     ins.set_train_epochs(args.num_epochs)
     ins.set_loss_type(args.loss_type)
+    ins.set_mixed_precision(True)
     # 1. Build a dali reader
     gpu_id = ins.trainer_id % 8 # Assume 8 card per machine
     dali_iter = dali.train(ins.train_batch_size,
-- 
GitLab