File tree 4 files changed +10
-4
lines changed
crates/burn-core/src/optim
4 files changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -172,7 +172,7 @@ mod tests {
172
172
BinFileRecorder :: < FullPrecisionSettings > :: default ( )
173
173
. record (
174
174
optimizer. to_record ( ) ,
175
- std:: env:: temp_dir ( ) . as_path ( ) . join ( "test_optim " ) ,
175
+ std:: env:: temp_dir ( ) . as_path ( ) . join ( "test_optim_adagrad " ) ,
176
176
)
177
177
. unwrap ( ) ;
178
178
Original file line number Diff line number Diff line change @@ -205,7 +205,7 @@ mod tests {
205
205
BinFileRecorder :: < FullPrecisionSettings > :: default ( )
206
206
. record (
207
207
optimizer. to_record ( ) ,
208
- std:: env:: temp_dir ( ) . as_path ( ) . join ( "test_optim " ) ,
208
+ std:: env:: temp_dir ( ) . as_path ( ) . join ( "test_optim_adam " ) ,
209
209
)
210
210
. unwrap ( ) ;
211
211
Original file line number Diff line number Diff line change @@ -217,7 +217,10 @@ mod tests {
217
217
let _linear = optimizer. step ( LEARNING_RATE , linear, grads) ;
218
218
let temp_dir = TempDir :: new ( ) . unwrap ( ) ;
219
219
BinFileRecorder :: < FullPrecisionSettings > :: default ( )
220
- . record ( optimizer. to_record ( ) , temp_dir. path ( ) . join ( "test_optim" ) )
220
+ . record (
221
+ optimizer. to_record ( ) ,
222
+ temp_dir. path ( ) . join ( "test_optim_adamw" ) ,
223
+ )
221
224
. unwrap ( ) ;
222
225
223
226
let state_optim_before = optimizer. to_record ( ) ;
Original file line number Diff line number Diff line change @@ -334,7 +334,10 @@ mod tests {
334
334
let _linear = optimizer. step ( LEARNING_RATE , linear, grads) ;
335
335
let temp_dir = TempDir :: new ( ) . unwrap ( ) ;
336
336
BinFileRecorder :: < FullPrecisionSettings > :: default ( )
337
- . record ( optimizer. to_record ( ) , temp_dir. path ( ) . join ( "test_optim" ) )
337
+ . record (
338
+ optimizer. to_record ( ) ,
339
+ temp_dir. path ( ) . join ( "test_optim_rmsprop" ) ,
340
+ )
338
341
. unwrap ( ) ;
339
342
340
343
let state_optim_before = optimizer. to_record ( ) ;
You can’t perform that action at this time.
0 commit comments