Skip to content

Virusdoll/Active-Negative-Loss

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Active-Negative-Loss-Functions

Requirements

python >= 3.9, torch >= 1.12.1, torchvision >= 0.13.1, numpy >= 1.23.1

How to use

Configs

Check '*.json' file in the config folder for each exeriment.

Arguments

  • gpu: GPU id
  • seed: random seed
  • config: config name
  • noise_type: 'sym' if use symmetric noise, 'asym' if use asymmetric noise
  • noise_rate: noise rate
  • eval_freq: frequency of evaluation, default is 1
  • tuning: use the tuning settings (90% of the original training set as training set and 10% as validation set)

Example

Training ANL-CE on CIFAR-10 with 0.8 symmetric noise:

python main.py \
--gpu 0 \
--seed 1 \
--config cifar10_anl_ce \
--noise_type sym \
--noise_rate 0.8 \
--eval_freq 10

Thanks

Moreover, we thank the codes implemented by Ma et al. and Zhou et al..

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages