UoA-CARES / cares_reinforcement_learning

CARES Reinforcement Learning Package
11 stars 2 forks source link

BtachNorm - Set Eval #222

Open beardyFace opened 2 days ago

beardyFace commented 2 days ago

To handle the generic use of Batchnorm, all algorithm updates need to set Eval/Train appropriately in the train methods