Skip to content

Commit 78c90fe

Browse files
committed
set up training
1 parent 786e997 commit 78c90fe

File tree

1 file changed

+14
-34
lines changed

1 file changed

+14
-34
lines changed

docs/tutorials/land_cover.ipynb

+14-34
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@
202202
},
203203
{
204204
"cell_type": "code",
205-
"execution_count": 12,
205+
"execution_count": 34,
206206
"id": "e5fd47f6-1b43-48af-b843-2ef30859a624",
207207
"metadata": {
208208
"scrolled": true
@@ -212,22 +212,7 @@
212212
"name": "stdout",
213213
"output_type": "stream",
214214
"text": [
215-
"Using downloaded and verified file: /data/datatorchgeo/HCAT2.csv\n",
216-
"Downloading https://zenodo.org/records/8229128/files/FR_2018.zip to /data/datatorchgeo/FR_2018.zip\n"
217-
]
218-
},
219-
{
220-
"name": "stderr",
221-
"output_type": "stream",
222-
"text": [
223-
"100%|██████████| 2629629542/2629629542 [00:15<00:00, 165140701.25it/s]\n"
224-
]
225-
},
226-
{
227-
"name": "stdout",
228-
"output_type": "stream",
229-
"text": [
230-
"Extracting /data/datatorchgeo/FR_2018.zip to /data/datatorchgeo\n",
215+
"Files already downloaded and verified\n",
231216
"Converting EuroCrops_ CRS from EPSG:4326 to EPSG:32631\n",
232217
"Converting EuroCrops_ res from 1e-05 to 10.0\n",
233218
"Files already downloaded and verified\n",
@@ -276,9 +261,9 @@
276261
"datamodule = Sentinel2EuroCropsDataModule(\n",
277262
" sentinel2_paths=\"/data/sentinel\",\n",
278263
" eurocrops_paths=\"/data/datatorchgeo\",\n",
279-
" batch_size=16,\n",
280-
" patch_size=64,\n",
281-
" num_workers=32,\n",
264+
" batch_size=64,\n",
265+
" patch_size=256,\n",
266+
" num_workers=64,\n",
282267
" eurocrops_download=True\n",
283268
")\n",
284269
"\n",
@@ -292,7 +277,7 @@
292277
},
293278
{
294279
"cell_type": "code",
295-
"execution_count": 13,
280+
"execution_count": 35,
296281
"id": "cfb04476-0d0e-48cb-b678-757cbcebe76b",
297282
"metadata": {
298283
"scrolled": true
@@ -311,7 +296,7 @@
311296
" ['/data/datatorchgeo/FR_2018/FR_2018_EC21.shp'])"
312297
]
313298
},
314-
"execution_count": 13,
299+
"execution_count": 35,
315300
"metadata": {},
316301
"output_type": "execute_result"
317302
}
@@ -675,7 +660,7 @@
675660
},
676661
{
677662
"cell_type": "code",
678-
"execution_count": 29,
663+
"execution_count": 36,
679664
"id": "33e40648-e1d6-4f0c-a4c4-4baec6ee8556",
680665
"metadata": {},
681666
"outputs": [
@@ -1079,7 +1064,7 @@
10791064
" '3399000000': 395}"
10801065
]
10811066
},
1082-
"execution_count": 29,
1067+
"execution_count": 36,
10831068
"metadata": {},
10841069
"output_type": "execute_result"
10851070
}
@@ -1100,19 +1085,14 @@
11001085
"name": "stderr",
11011086
"output_type": "stream",
11021087
"text": [
1103-
"INFO:torch.distributed.nn.jit.instantiator:Created a temporary directory at /tmp/tmpe85uwi0w\n",
1104-
"INFO:torch.distributed.nn.jit.instantiator:Writing /tmp/tmpe85uwi0w/_remote_module_non_scriptable.py\n",
11051088
"INFO: GPU available: True (cuda), used: False\n",
11061089
"INFO:lightning.pytorch.utilities.rank_zero:GPU available: True (cuda), used: False\n",
11071090
"INFO: TPU available: False, using: 0 TPU cores\n",
11081091
"INFO:lightning.pytorch.utilities.rank_zero:TPU available: False, using: 0 TPU cores\n",
11091092
"INFO: IPU available: False, using: 0 IPUs\n",
11101093
"INFO:lightning.pytorch.utilities.rank_zero:IPU available: False, using: 0 IPUs\n",
11111094
"INFO: HPU available: False, using: 0 HPUs\n",
1112-
"INFO:lightning.pytorch.utilities.rank_zero:HPU available: False, using: 0 HPUs\n",
1113-
"/opt/conda/envs/ood/lib/python3.11/site-packages/lightning/pytorch/trainer/setup.py:187: GPU available but not used. You can set it by doing `Trainer(accelerator='gpu')`.\n",
1114-
"INFO: Running in `fast_dev_run` mode: will run the requested loop using 1 batch(es). Logging and checkpointing is suppressed.\n",
1115-
"INFO:lightning.pytorch.utilities.rank_zero:Running in `fast_dev_run` mode: will run the requested loop using 1 batch(es). Logging and checkpointing is suppressed.\n"
1095+
"INFO:lightning.pytorch.utilities.rank_zero:HPU available: False, using: 0 HPUs\n"
11161096
]
11171097
},
11181098
{
@@ -1166,7 +1146,7 @@
11661146
{
11671147
"data": {
11681148
"application/vnd.jupyter.widget-view+json": {
1169-
"model_id": "9b2c8c920aec4933824dfccd37137f32",
1149+
"model_id": "16c2da485c104d8d8b5b485e219f4861",
11701150
"version_major": 2,
11711151
"version_minor": 0
11721152
},
@@ -1204,11 +1184,11 @@
12041184
"trainer = Trainer(\n",
12051185
" accelerator='cpu',\n",
12061186
" default_root_dir='./',\n",
1207-
" fast_dev_run=True,\n",
1208-
" log_every_n_steps=1,\n",
1187+
" fast_dev_run=False,\n",
1188+
" log_every_n_steps=10,\n",
12091189
" min_epochs=2,\n",
12101190
" max_epochs=20,\n",
1211-
")\n",
1191+
" num_sanity_val_steps=0)\n",
12121192
"\n",
12131193
"trainer.fit(model=task, datamodule=datamodule)"
12141194
]

0 commit comments

Comments
 (0)