Assign requires shapes of both tensors to match. lhs shape= [13] rhs shape= [150]
I have change restore_var = tf.global_variables() to restore_var = [v for v in tf.global_variables() if 'conv6_cls' not in v.name] Update INPUT_SIZE to '512, 512' Update NUM_CLASSES to 13 filter_scale=2. model_type = 'others' when run python train.py --dataset others ,I get erros: raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [13] rhs shape= [150] [[node save/Assign_382 (defined at /home/sherry/cuimiao/Fabric_defect_detection/ICNet-tensorflow/network.py:79) ]]
Errors may have originated from an input operation. Input Source operations connected to node save/Assign_382: sub24_out/biases (defined at /home/sherry/cuimiao/Fabric_defect_detection/ICNet-tensorflow/network.py:145)
Original stack trace for 'save/Assign_382':
File "train.py", line 167, in