Preparing for tpus branch merge

spaceml-org · May 1, 2024 · 711f487 · 711f487
1 parent 4d91860
commit 711f487
Show file tree

Hide file tree

Showing 4 changed files with 1,735 additions and 69 deletions.
diff --git a/experiments/pretrain_32.2M_samae.yaml b/experiments/pretrain_32.2M_samae.yaml
@@ -46,8 +46,8 @@ experiment:
 
 # dataset configuration
 data:
-  min_date: "0000-00-00 00:00:00" # NOT IMPLEMENTED # minimum is '2010-09-09 00:00:11.08'
-  max_date: "0000-00-00 00:00:00" # NOT IMPLEMENTED # maximum is '2023-05-26 06:36:08.072'
+  min_date: '2010-09-09 00:00:11.08' # NOT IMPLEMENTED # minimum is '2010-09-09 00:00:11.08'
+  max_date: '2023-05-26 06:36:08.072' # NOT IMPLEMENTED # maximum is '2023-05-26 06:36:08.072'
   month_splits: # non selected months will form training set 
     # train: [1,2,3,4,5,6,7,8,9,10]
     val: [11]

diff --git a/notebooks/test_finetuning.ipynb b/notebooks/test_finetuning.ipynb