You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by zh...@apache.org on 2022/12/02 09:06:28 UTC
[singa] branch dev updated: Disable dataset shape checking for cnn example
This is an automated email from the ASF dual-hosted git repository.
zhaojing pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/singa.git
The following commit(s) were added to refs/heads/dev by this push:
new 7e35a2e6 Disable dataset shape checking for cnn example
new 1b805f70 Merge pull request #1017 from zlheui/disable-dataset-shape-checking
7e35a2e6 is described below
commit 7e35a2e6fe53f99dafccae2472e41adb3efe50c5
Author: zhulei <zl...@gmail.com>
AuthorDate: Fri Dec 2 16:30:26 2022 +0800
Disable dataset shape checking for cnn example
---
examples/cnn/train_cnn.py | 7 -------
1 file changed, 7 deletions(-)
diff --git a/examples/cnn/train_cnn.py b/examples/cnn/train_cnn.py
index 20d5c444..5c064707 100644
--- a/examples/cnn/train_cnn.py
+++ b/examples/cnn/train_cnn.py
@@ -165,13 +165,6 @@ def run(global_rank,
train_x, train_y, val_x, val_y = partition(global_rank, world_size,
train_x, train_y, val_x,
val_y)
- '''
- # check dataset shape correctness
- if global_rank == 0:
- print("Check the shape of dataset:")
- print(train_x.shape)
- print(train_y.shape)
- '''
if model.dimension == 4:
tx = tensor.Tensor(