How to make 3 Fully connected layer

gwding / draw_convnet

2.64k stars 513 forks source link

How to make 3 Fully connected layer #15

Open cir17lc opened 3 years ago

cir17lc commented 3 years ago

Hi, thanks for the code. However for my problem I have 3 fully connected layer, however when change size_list = [(fc_unit_size, fc_unit_size)] * 4, num_list = [4500, 4500,4500, 1148] I have this error _IndexError: index 3 is out of bounds for axis 0 with size 3

Do you know how to overcome this ? Thanks !

poxiao2 commented 2 years ago

I have the same question, have you solved it?