fix: update test expectations for batch_size=16384
This commit is contained in:
@@ -1096,7 +1096,7 @@ mod tests {
|
||||
|
||||
// [training] values must have been applied
|
||||
assert_eq!(hp.epochs, 200);
|
||||
assert_eq!(hp.batch_size, 8192);
|
||||
assert_eq!(hp.batch_size, 16384);
|
||||
assert!((hp.learning_rate - 0.0001).abs() < 1e-10);
|
||||
|
||||
// OFI MBP-10 data dir should be set in production profile
|
||||
@@ -1364,7 +1364,7 @@ mod tests {
|
||||
|
||||
// [training] section
|
||||
assert_eq!(hp.epochs, 200);
|
||||
assert_eq!(hp.batch_size, 8192);
|
||||
assert_eq!(hp.batch_size, 16384);
|
||||
assert!((hp.learning_rate - 0.0001).abs() < 1e-10);
|
||||
assert!((hp.gamma - 0.99).abs() < 0.001);
|
||||
assert!((hp.reward_scale - 1.0).abs() < 0.01);
|
||||
|
||||
Reference in New Issue
Block a user