folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

check train test (#111)

Browse Source 
* Update setup.py
* scheduler for all torch versions
---------

Co-authored-by: Dario Coscia <dariocoscia@cli-10-110-0-208.WIFIeduroamSTUD.units.it>
Co-authored-by: Nicola Demo <demo.nicola@gmail.com>
This commit is contained in:
Dario Coscia
2023-06-13 12:07:04 +02:00
committed by Nicola Demo
parent 8ce3405edc
commit 0fb93a73ab
3 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -15,7 +15,7 @@ VERSION = meta['__version__']
KEYWORDS = 'physics-informed neural-network'
REQUIRED = [
'numpy', 'matplotlib', 'torch'
'numpy', 'matplotlib', 'torch', 'lightning'
]
EXTRAS = {