search

13o-bbr-bbq / machine_learning_security

Source code about machine learning and security.
1.96k stars 648 forks source link

When running test mode this error #14

Closed cybert79 closed 6 years ago

cybert79 commented 6 years ago

This is on another VM (VMWorkstation) Kali linux, I get this error

Traceback (most recent call last): File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1322, in _do_call return fn(*args) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1307, in _run_fn options, feed_dict, fetch_list, target_list, run_metadata) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1409, in _call_tf_sessionrun run_metadata) tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [200,539] rhs shape= [200,504] [[Node: save/Assign_9 = Assign[T=DT_FLOAT, _class=["loc:@local_thread1/dense_9/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](local_thread1/dense_9/kernel, save/RestoreV2:9)]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "DeepExploit.py", line 1994, in saver.restore(SESS, env.save_file) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1802, in restore {self.saver_def.filename_tensor_name: save_path}) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 900, in run run_metadata_ptr) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1135, in _run feed_dict_tensor, options, run_metadata) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1316, in _do_run run_metadata) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1335, in _do_call raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [200,539] rhs shape= [200,504] [[Node: save/Assign_9 = Assign[T=DT_FLOAT, _class=["loc:@local_thread1/dense_9/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](local_thread1/dense_9/kernel, save/RestoreV2:9)]]

Caused by op 'save/Assign_9', defined at: File "DeepExploit.py", line 1971, in saver = tf.train.Saver() File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1338, in init self.build() File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1347, in build self._build(self._filename, build_save=True, build_restore=True) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1384, in _build build_save=build_save, build_restore=build_restore) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 835, in _build_internal restore_sequentially, reshape) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 494, in _AddRestoreOps assign_ops.append(saveable.restore(saveable_tensors, shapes)) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 185, in restore self.op.get_shape().is_fully_defined()) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/ops/state_ops.py", line 283, in assign validate_shape=validate_shape) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/ops/gen_state_ops.py", line 60, in assign use_locking=use_locking, name=name) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper op_def=op_def) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3392, in create_op op_def=op_def) File "/home/drwho/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1718, in init self._traceback = self._graph._extract_stack() # pylint: disable=protected-access

InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [200,539] rhs shape= [200,504] [[Node: save/Assign_9 = Assign[T=DT_FLOAT, _class=["loc:@local_thread1/dense_9/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](local_thread1/dense_9/kernel, save/RestoreV2:9)]]

13o-bbr-bbq commented 6 years ago

when you change to other VM (other Metasploit version), you don't use old trained file (your Deep Exploit's path/trained_data/*).
because, if number of payloads changes, structure of neural network changes within the Deep Exploit.

this is my important task. i'll upcoming improve.