Saving weights and logs of step 1000

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "roberta-base",
+  "_name_or_path": "./",
   "architectures": [
     "RobertaForMaskedLM"
   ],

events.out.tfevents.1642699985.t1v-n-eedfb410-w-0.2460473.0.v2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:162bf46ab70102e8a81c3d59c1363eaa2873cd9bb479546a2569c5e28390020a
+size 147136

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e6c1b729fa1c3dd64fbd625887667b363143278ab0161b5aeb411a6811b37680
+oid sha256:2610909ba986dd96626aa623d3bf3c3d78be05bbb07bf25b595aa95d0ef29c04
 size 498796983

run_mlm_flax.py

@@ -396,7 +396,7 @@ def main():
     # download the dataset.
     if data_args.dataset_name is not None:
         # Downloading and loading a dataset from the hub.
-        datasets = load_dataset(data_args.dataset_name, data_args.dataset_config_name, cache_dir=model_args.cache_dir)
+        datasets = load_dataset(data_args.dataset_name, data_args.dataset_config_name, cache_dir=model_args.cache_dir, use_auth_token=True)
 
         if "validation" not in datasets.keys():
             datasets["validation"] = load_dataset(