Incorrect Weight Parameter Used for Loss Calculation

[elTorado]

File: train.py

Issue:
Incorrect weight parameter options['beta'] used in loss calculations.
Lines affected: 93 and 117.

Expected:
Use options['--weight-pl'] for weight in the loss calculations, as it's intended for center loss, not options['beta'], which is for entropy loss.

Details:

• Line 93: generator_loss calculation.
• Line 117: total_loss calculation.

Suggested Fix:
Replace options['beta'] with options['--weight-pl'] in the affected lines.