-
Notifications
You must be signed in to change notification settings - Fork 7
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Version of pytorch-lightning #4
Comments
Hi, we train our model by PyTorch-lightning 1.8.0 previously, but I tested my model in latest version of PyTorch-lightning (v2.x) before I release my code. I think it's ok to use this code. Could you share your problems encountered? |
Thanks for your reply! I have run the code successfully after I commented the Before that, the problem is shown as the following screenshot: |
Thank you for sharing! |
Hi! I have found one possible solution. The problem may come from the configuration of using The original version (L31-L32, L226) is: from pytorch_lightning.loggers import WandbLogger,TensorBoardLogger
logger = TensorBoardLogger(r'tb_logs', name='udrs2former') trainer_defaults = {'devices':[1,2,3],'callbacks':[checkpoint_callback,lr_monitor],'logger':logger} The first block may be removed and the second can be altered as follows: train_defaults = {
# configs of device and callbacks goes here...
'logger': {
'class_path': 'lightning.pytorch.loggers.TensorBoardLogger',
'init_args': {
'save_dir': 'tb_logs',
'name': 'udrs2former'
}
}
} |
This may be caused by the updated version. This is indeed a good solution. Thank you. |
没有~ |
我也是这个问题,我打算降一下pytorch版本了 |
可以试试~ |
Hi! It is a great and intersting work. But I am wondering the version of pytorch-lightning you used because I met some problem with the latest version (v2.x)
The text was updated successfully, but these errors were encountered: