issues
search
google
/
uncertainty-baselines
High-quality implementations of standard and SOTA methods on a variety of tasks.
Apache License 2.0
1.46k
stars
203
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Add general SKAI dataset.
#1280
copybara-service[bot]
closed
1 year ago
0
Update SKAI dataset code.
#1279
copybara-service[bot]
closed
1 year ago
0
Saves best model during training. Updates default behavior to saving model (more easily reproducible results) instead of checkpoints.
#1278
copybara-service[bot]
closed
1 year ago
1
Update types to prevent pytype errors with an upcoming change to JAX.
#1277
copybara-service[bot]
closed
1 year ago
0
Internal
#1276
copybara-service[bot]
closed
1 year ago
1
Suppress some pytype errors related to improved JAX types under pytype.
#1275
copybara-service[bot]
closed
1 year ago
0
Open source CIFAR-10H Tensorflow Dataset construction and preprocessing.
#1274
copybara-service[bot]
closed
1 year ago
0
plex_vit_demo : Failed to get url-imagenetv2public
#1273
Eezzeldin
opened
1 year ago
0
Can ignore some directory in t5 utils `latest_checkpoint_paths`.
#1272
copybara-service[bot]
closed
1 year ago
0
Add two_tower_model.
#1271
copybara-service[bot]
opened
1 year ago
1
Assignment map for T5 het in the diagonal case.
#1270
copybara-service[bot]
closed
1 year ago
0
Replaces references to jax.numpy.DeviceArray with jax.Array.
#1269
copybara-service[bot]
closed
1 year ago
0
Update T5 het gin config.
#1268
copybara-service[bot]
closed
1 year ago
0
Silence some pytype errors.
#1267
copybara-service[bot]
closed
1 year ago
0
Add covariance initial scale as argument of `HeteroscedasticDecoder`.
#1266
copybara-service[bot]
closed
1 year ago
0
The heteroscedastic layer can reuse the pretrained head of T5 models.
#1265
copybara-service[bot]
closed
1 year ago
0
Suppress some pytype errors related to jnp.DeviceArray == jax.Array.
#1264
copybara-service[bot]
closed
1 year ago
0
Add in T5 the learnable temperature parameter of the het layer.
#1263
copybara-service[bot]
closed
1 year ago
0
Surface more arguments of the `HeteroscedasticDecoder`.
#1262
copybara-service[bot]
closed
1 year ago
0
[JAX] Fix up uses of PyTree and PyTreeDef types.
#1261
copybara-service[bot]
closed
1 year ago
0
input_utils to importing in Colab notebook
#1260
JonoSax
opened
1 year ago
0
Update default behavior.
#1259
copybara-service[bot]
closed
1 year ago
1
#ISP Fix bugs in toxic_comment dataset argument passing, evaluation and inference
#1258
copybara-service[bot]
closed
1 year ago
0
Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
#1257
copybara-service[bot]
closed
1 year ago
0
Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.
#1256
copybara-service[bot]
closed
1 year ago
0
Silence some pytype errors.
#1255
copybara-service[bot]
closed
1 year ago
0
Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.
#1254
copybara-service[bot]
closed
1 year ago
0
#toxic_comments support filter data by created date
#1253
copybara-service[bot]
closed
1 year ago
0
Question about SNGP BERT flags
#1252
xuefei-wang
closed
1 year ago
2
Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.
#1251
copybara-service[bot]
closed
1 year ago
0
Create a T5x experiment setup for BIG-Bench Multiple Choice Question Answering.
#1250
copybara-service[bot]
closed
1 year ago
0
Internal migration of Xmanager API.
#1249
copybara-service[bot]
opened
1 year ago
1
Fix Imagenet ReaL masking bug.
#1248
copybara-service[bot]
closed
1 year ago
0
Internal.
#1247
copybara-service[bot]
closed
1 year ago
0
Implements upsampling for the most underrepresented group.
#1246
copybara-service[bot]
closed
1 year ago
1
Calculates weighted average accuracy, weighting each subgroup's acc by its proportionality.
#1245
copybara-service[bot]
closed
1 year ago
1
Update pytype disable markers for Python 3.10.
#1244
copybara-service[bot]
closed
1 year ago
0
Minor modififcations to CIFAR PI scripts.
#1243
copybara-service[bot]
closed
1 year ago
0
Fix imports in CIFAR-PI baselines
#1242
gortizji
opened
1 year ago
0
Upgrade python from 3.7 to 3.8 in the CI to address failures from features from newer versions.
#1241
copybara-service[bot]
opened
1 year ago
0
Adds support for hyperparameter sweeping with Vizier. Also adds a flag to control whether or not to make adjustments to the TF Keras ResNet model to be more like the PyTorch ResNet model.
#1240
copybara-service[bot]
closed
1 year ago
1
Catch and warn for import errors from TFDSBuilderFromSQLClientData.
#1239
copybara-service[bot]
closed
1 year ago
0
Fix types of ImageNet ReaL preprocessing.
#1238
copybara-service[bot]
closed
1 year ago
0
Zero-shot Clip Robustness and generalization Evaluation
#1237
copybara-service[bot]
closed
1 year ago
1
Open source CIFAR-10N and CIFAR-100N privileged information datasets and no PI and TRAM baselines.
#1236
copybara-service[bot]
closed
1 year ago
0
RETINA: Data Loading Improvements, TF Cast Fix
#1235
nband
opened
1 year ago
1
Adds callback to log metrics in XManager to make it easier to identify best hyperparam values.
#1234
copybara-service[bot]
closed
1 year ago
1
Expands hyperparameter sweep to include optimizer types and number of epochs. Adds example command.
#1233
copybara-service[bot]
closed
1 year ago
1
Checkpoint evaluation when processing multiple model checkpoints.
#1232
copybara-service[bot]
closed
1 year ago
0
[NumPy] Remove references to deprecated NumPy type aliases.
#1231
copybara-service[bot]
closed
1 year ago
0
Previous
Next