-
Notifications
You must be signed in to change notification settings - Fork 1
Description
After configuring the environment from the original file, directly run 'python main. py' to report an error, Could you give me some suggestions?
...........................................................................................................................................
PS C:\Code\Project\supervised_layout_benchmark-main> python main.py
GPU available: True, used: True
TPU available: None, using: 0 TPU cores
LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]
Namespace(backbone='AlexNet', batch_size=2, boundary='one_point', config=False, data_format='mat', data_root='samples/data/', gpus=1, input_size=200, lr=0.001, max_epochs=50, mean_heat=298.0, mean_layout=0.0, metric='mae_boundary', mode='train', model_name='FCN', num_workers=4, profiler=False, resume_from_checkpoint=None, seed=1, std_heat=50.0, std_layout=1000.0, test_args=False, test_check_num='0', test_list='test/test_0.txt', train_list='train/train_val.txt', train_size=0.8, use_16bit=False, val_check_interval=1.0)
Prepared dataset, train:8,val:2, test:8
Traceback (most recent call last):
File "main.py", line 65, in
main()
File "main.py", line 61, in main
getattr(eval(hparams.mode), "main")(hparams)
File "C:\Code\Project\supervised_layout_benchmark-main\src\train.py", line 46, in main
trainer.fit(model)
File "C:\Users\lenovo\AppData\Roaming\Python\Python38\site-packages\pytorch_lightning\trainer\trainer.py", line 454, in fit
self.accelerator_backend.setup(model)
File "C:\Users\lenovo\AppData\Roaming\Python\Python38\site-packages\pytorch_lightning\accelerators\gpu_accelerator.py", line 52, in setup
self.setup_optimizers(model)
File "C:\Users\lenovo\AppData\Roaming\Python\Python38\site-packages\pytorch_lightning\accelerators\accelerator.py", line 145, in setup_optimizers
optimizers, lr_schedulers, optimizer_frequencies = self.trainer.init_optimizers(model)
File "C:\Users\lenovo\AppData\Roaming\Python\Python38\site-packages\pytorch_lightning\trainer\optimizers.py", line 78, in init_optimizers
lr_schedulers = self.configure_schedulers(lr_schedulers, monitor=monitor)
File "C:\Users\lenovo\AppData\Roaming\Python\Python38\site-packages\pytorch_lightning\trainer\optimizers.py", line 136, in configure_schedulers
raise ValueError(f'The provided lr scheduler "{scheduler}" is invalid')
ValueError: The provided lr scheduler "<torch.optim.lr_scheduler.ExponentialLR object at 0x000001A01AFB96A0>" is invalid