2 errors while reproducing the results

[Angelina1996]

Hi Rahul,
Thanks for such excellent work, could you please resolve the following 2 errors:

File "main.py", line 191, in
training_model.accumulate(phase="train", save=args.save_features)
TypeError: accumulate() got an unexpected keyword argument 'save'
wandb: Waiting for W&B process to finish... (failed 1).

The second error is, it cant find the Pretrain path :

===> Saving cfg parameters to: /data/angelina/Experimentaltesults/TailCalibration/logs/vit/CIFAR100_LT/100/CBD+TailCalibX_CIFAR100_LT_imb_100_resnet32/cfg.yaml
Using steps for training.
Loading Softmax Loss.
Using 1 GPUs.
Loading Cosine Dot Product Classifier.
Traceback (most recent call last):
File "main.py", line 171, in
training_model = source_import(config["core"]).get_core(config, data)
File "./libs/core/TailCalibX.py", line 311, in get_core
return model(*args)
File "/home/angelina/ TailCalibration/TailCalib_Feature_Generation/libs/core/core_base.py", line 54, in init
self.init_models()
File "/home/angelina/ TailCalibration /TailCalib_Feature_Generation/libs/core/core_base.py", line 76, in init_models
self.networks[key] = source_import(def_file).create_model(**model_args)
File "./libs/models/CosineDotProductClassifier.py", line 48, in create_model
raise Exception(f"Pretrain path doesn't exist!!-{pretrain_dir}")
Exception: Pretrain path doesn't exist!!-/data/angelina/TailCalibration/logs/vit/CIFAR100_LT/100/CBD_CIFAR100_LT_imb_100_resnet32/alpha_0.8,beta_100.0_normal_k_1_aug_k_0/final_model_checkpoint.pth
wandb: Waiting for W&B process to finish... (failed 1).