MCPcopy Create free account
hub / github.com/Kaggle/docker-python / test_cuda_nn

Method test_cuda_nn

tests/test_pytorch.py:39–46  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

37 # These throw if cuda is misconfigured
38 tnn.GRUCell(10,10).cuda()
39 tnn.RNNCell(10,10).cuda()
40 tnn.LSTMCell(10,10).cuda()
41 tnn.GRU(10,10).cuda()
42 tnn.LSTM(10,10).cuda()
43 tnn.RNN(10,10).cuda()

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected