search

B-cos / B-cos-v2

Official PyTorch implementation of improved B-cos models
Apache License 2.0
39 stars 5 forks source link

Update requirements-train.txt #5

Closed mhmoodlan closed 10 months ago

mhmoodlan commented 10 months ago

Limit pytorch-lightning version to be less than 2.0.0, as versions from 2.0.0 upwards seem to be incompatible. Specifically, it breaks when running training script, e.g.: python train.py --dataset CIFAR10 --base_network norm_ablations_final --experiment_name resnet_56-nomaxout

output:

TypeError: ClassificationLitModel.configure_gradient_clipping() missing 1 required positional argument: 'optimizer_idx'

nps1ngh commented 10 months ago

Hi,

great catch! Thanks a lot! :smile: