csyanbin / tpn Goto Github PK
View Code? Open in Web Editor NEWCode for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.
Code for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.
Have you ever met this error. It appears when I run the code.
train epoc:0: 0%| | 0/100 [00:00<?, ?it/s]2019-04-08 19:45:32.987229: I tensorflow/core/kernels/cuda_solvers.cc:159] Creating CudaSolver handles for stream 0x1564c380
2019-04-08 19:45:33.117242: W tensorflow/core/framework/op_kernel.cc:1318] OP_REQUIRES failed at matrix_inverse_op.cc:223 : Internal: tensorflow/core/kernels/cuda_solvers.cc:408: cuSolverDN call failed with status =6
Traceback (most recent call last):
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1322, in _do_call
return fn(*args)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1307, in _run_fn
options, feed_dict, fetch_list, target_list, run_metadata)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1409, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.InternalError: tensorflow/core/kernels/cuda_solvers.cc:408: cuSolverDN call failed with status =6
[[Node: MatrixInverse = MatrixInverseT=DT_FLOAT, adjoint=false, _device="/job:localhost/replica:0/task:0/device:GPU:0"]]
[[Node: Mean_2/_51 = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_2112_Mean_2", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/media/hlf/51aa617b-dbbc-4ad1-af81-45cf8dfce172/hlf/code/TPN-master/train.py", line 213, in
_, summaries, step, ls, ac = sess.run([train_op, train_summary_op, global_step, ce_loss, acc], feed_dict={m.x: support, m.ys:s_labels, m.q: query, m.y:q_labels, m.phase:1})
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 900, in run
run_metadata_ptr)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1135, in _run
feed_dict_tensor, options, run_metadata)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1316, in _do_run
run_metadata)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 1335, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InternalError: tensorflow/core/kernels/cuda_solvers.cc:408: cuSolverDN call failed with status =6
[[Node: MatrixInverse = MatrixInverseT=DT_FLOAT, adjoint=false, _device="/job:localhost/replica:0/task:0/device:GPU:0"]]
[[Node: Mean_2/_51 = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_2112_Mean_2", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]
Caused by op 'MatrixInverse', defined at:
File "/media/hlf/51aa617b-dbbc-4ad1-af81-45cf8dfce172/hlf/code/TPN-master/train.py", line 148, in
ce_loss,acc,sigma_value = m.construct()
File "/media/hlf/51aa617b-dbbc-4ad1-af81-45cf8dfce172/hlf/code/TPN-master/models.py", line 88, in construct
ce_loss, acc, sigma_value = self.label_prop(emb_x, emb_q, ys_one_hot)
File "/media/hlf/51aa617b-dbbc-4ad1-af81-45cf8dfce172/hlf/code/TPN-master/models.py", line 139, in label_prop
F = tf.matrix_inverse(tf.cast(F, tf.float32))
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/ops/gen_linalg_ops.py", line 1049, in matrix_inverse
"MatrixInverse", input=input, adjoint=adjoint, name=name)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 3392, in create_op
op_def=op_def)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 1718, in init
self._traceback = self._graph._extract_stack() # pylint: disable=protected-access
InternalError (see above for traceback): tensorflow/core/kernels/cuda_solvers.cc:408: cuSolverDN call failed with status =6
[[Node: MatrixInverse = MatrixInverseT=DT_FLOAT, adjoint=false, _device="/job:localhost/replica:0/task:0/device:GPU:0"]]
[[Node: Mean_2/_51 = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_2112_Mean_2", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]
I've found that Wii=1 , rather than 0 , will it influence a lot?
Hello,
Very interesting paper. Thank you for uploading the code. In the paper the results of applying transductive idea on MAML, Reptile and RelationNet algorithms are reported in table1 and table2. Is there a way to run MAML with transductive in this repository?
Hi!
I cannot understand the sentence, 'Transductive methods directly use test set as unlabeled data while semi-supervised learning usually has an extra unlabeled set.', in Sec. 4.4.
Can I obtain your code for Semi-TPN.
I am looking forward to your reply!
Thank you very much!
Hello, it's really a nice work. And i found when 5*1 test samples, the model has a good performance. But i wonder if there is only one query sample, does the model has a recognize ability and how to test it?
Thank you for your nice work. I had a problem while browsing the code.
What does the 1600 dimension in this line of code mean and why is it set to this number?
Line 109 in 3bc9eea
Traceback (most recent call last):
File "test.py", line 122, in
loader_test.load_data_pkl()
File "/home/x_y/lyf/TPN/dataset_mini.py", line 86, in load_data_pkl
data = pkl.load(f)
_pickle.UnpicklingError: invalid load key, '*'.
hello,could you help me?
A declarative, efficient, and flexible JavaScript library for building user interfaces.
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. ๐๐๐
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google โค๏ธ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.