|
|
|
@ -33,7 +33,7 @@ import numpy as np
|
|
|
|
|
from config_imagenet import config
|
|
|
|
|
from crossentropy import CrossEntropy
|
|
|
|
|
from dataset_imagenet import create_dataset
|
|
|
|
|
from lr_generator import get_lr, warmup_cosine_annealing_lr
|
|
|
|
|
from lr_generator import warmup_cosine_annealing_lr
|
|
|
|
|
|
|
|
|
|
random.seed(1)
|
|
|
|
|
np.random.seed(1)
|
|
|
|
|