dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix unstable tests when run concurrently (#1724)

This commit is contained in:
Anton Blomström 2024-05-05 22:27:42 +02:00 committed by GitHub
parent 152509c378
commit f8994e044c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
crates/burn-core/src/optim/adagrad.rs
View File

@ -172,7 +172,7 @@ mod tests {
BinFileRecorder::<FullPrecisionSettings>::default()
.record(
optimizer.to_record(),
std::env::temp_dir().as_path().join("test_optim"),
std::env::temp_dir().as_path().join("test_optim_adagrad"),
)
.unwrap();

2
crates/burn-core/src/optim/adam.rs
View File

@ -205,7 +205,7 @@ mod tests {
BinFileRecorder::<FullPrecisionSettings>::default()
.record(
optimizer.to_record(),
std::env::temp_dir().as_path().join("test_optim"),
std::env::temp_dir().as_path().join("test_optim_adam"),
)
.unwrap();

5
crates/burn-core/src/optim/adamw.rs
View File

@ -217,7 +217,10 @@ mod tests {
let _linear = optimizer.step(LEARNING_RATE, linear, grads);
let temp_dir = TempDir::new().unwrap();
BinFileRecorder::<FullPrecisionSettings>::default()
.record(optimizer.to_record(), temp_dir.path().join("test_optim"))
.record(
optimizer.to_record(),
temp_dir.path().join("test_optim_adamw"),
)
.unwrap();
let state_optim_before = optimizer.to_record();

5
crates/burn-core/src/optim/rmsprop.rs
View File

@ -334,7 +334,10 @@ mod tests {
let _linear = optimizer.step(LEARNING_RATE, linear, grads);
let temp_dir = TempDir::new().unwrap();
BinFileRecorder::<FullPrecisionSettings>::default()
.record(optimizer.to_record(), temp_dir.path().join("test_optim"))
.record(
optimizer.to_record(),
temp_dir.path().join("test_optim_rmsprop"),
)
.unwrap();
let state_optim_before = optimizer.to_record();