Table 3 Experimental variable definition and settings.

From: Gearbox fault diagnosis method based on lightweight channel attention mechanism and transfer learning

Variables

Definitions

Values

Batch Size

Number of samples need to be processed per batch

32

Initial learning rate

Learning rate setting at the initial training

0.01

Learning rate decay

Automatically adjust learning rate by a decrease of 0.1

0.1

Dropout

Randomly delete neurons

0.2