nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[BUG] Efficientnet-b3 give negative lr #86

Closed internnos closed 3 years ago

internnos commented 3 years ago

Describe the bug Efficientnet-b3 give negative lr for first few epochs on cifar-10 dataset Navigate here to see the comet-ml log

To Reproduce Steps to reproduce the behavior: Below is the experiment configuration

experiment_name: efficientnet_b3_softmax_cifar10
logging: None,
  pytz_timezone: Asia/Jakarta
}
checkpoint: None
device: 'cuda:0'
output_directory: experiments/outputs
dataset: {
  train: {
    name: CIFAR10,
    args: {
      root: external/datasets,
      train: True,
      download: True
    },
  },
  eval: {
    name: CIFAR10,
    args: {
      root: external/datasets,
      train: False,
      download: True
    }
  },
}
dataloader: {
  module: PytorchDataLoader,
  args: {
    num_workers: 0,
    batch_size: 256,
    shuffle: True,
  },
}
model: {
  name: softmax,
  network_args: {
    backbone: efficientnet_b3,
    n_classes: 10,
    pretrained_backbone: True,
  },
  preprocess_args: {
    input_size: 32,
    input_normalization: {
      mean: [0.4914, 0.4822, 0.4465],
      std: [0.2023, 0.1994, 0.2010],
      scaler: 255,
    }
  },
  loss_args: {
    reduction: mean
  },
  postprocess_args: {}
}
trainer: {
  optimizer: {
    method: SGD,
    args: {
      lr: 0.0263,
      momentum: 0.9,
      weight_decay: 0.0005,
    }
  },
  lr_scheduler : {
    method : CosineLRWithWarmUp,
    args : {
      t_initial : 20,
      t_mul : 1.0,
      lr_min : 0.00001,
      warmup_lr_init: 0.00001,
      warmup_t: 2,
      cycle_limit : 1,
      t_in_epochs : True,
      decay_rate : 0.1,
    }
  },
  driver: {
    module: DefaultTrainer,
    args: {}
  },
  epoch: 20,
  save_epoch: 5
}
validator: {
  args: {},
  val_epoch: 4,
}
exporter : {
  module : onnx,
  args : {
    opset_version : 11,
  },
}

Expected behavior Lr > 0

Screenshots

Additional context Add any other context about the problem here.

triwahyuu commented 3 years ago

check my PR