fix

d772d390 · Minjie Wang · a8a4fcba · d772d390 · d772d390
Commit d772d390 authored Sep 25, 2018 by Minjie Wang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

CMakeLists.txt CMakeLists.txt +1 -1

examples/pytorch/tree_lstm/train.py examples/pytorch/tree_lstm/train.py +1 -1

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
 ########################################
 # Borrowed and adapted from TVM project
 ########################################
-cmake_minimum_required(VERSION 3.2)
+cmake_minimum_required(VERSION 2.8)
 project(dgl C CXX)

 # Utility functions

--- a/examples/pytorch/tree_lstm/train.py
+++ b/examples/pytorch/tree_lstm/train.py
@@ -37,7 +37,7 @@ def main(args):
        bg = dgl.batch(trees)
        if cuda:
            reprs = bg.get_n_repr()
-            reprs = {key : reprs[key].cuda()}
+            reprs = {key : val.cuda() for key, val in reprs.items()}
            bg.set_n_repr(reprs)
        return bg
    trainset = data.SST()