Wafaa_Wardah:

self.fc1 = nn.Linear(C_out_2 * H_out_2 * W_out_2, C_out_2 * H_out_2 * W_out_2) self.fc1 = nn.Linear(C_out_2 * H_out_2 * W_out_2, 2)


You don’t need the first self.fc1 statement. Otherwise, I think it is fine.