We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
iters_per_epoch
1 parent d151eaa commit 6a2a99fCopy full SHA for 6a2a99f
examples/resnet_finetune/src/main.rs
@@ -276,7 +276,7 @@ pub fn train<B: AutodiffBackend>(args: &Args) -> anyhow::Result<()> {
276
.num_workers(args.num_workers)
277
.build(valid);
278
279
- let iters_per_epoch = train_set_size / (args.batch_size * args.grads_accumulation);
+ let iters_per_epoch = train_set_size / args.batch_size;
280
let lr_scheduler = CosineAnnealingLrSchedulerConfig::new(
281
args.learning_rate,
282
iters_per_epoch * args.num_epochs,
0 commit comments