Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar10-cifar100.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar10_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar10-cifar100.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar10
pretrain_only: false
seed: 42
task_dataset_name: cifar100
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.5028
best_val_epoch: 19
test_acc: 0.496
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar10-kmnist.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar10_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar10-kmnist.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar10
pretrain_only: false
seed: 42
task_dataset_name: kmnist
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.9758333333333333
best_val_epoch: 14
test_acc: 0.9752
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar10-letters.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar10_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar10-letters.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar10
pretrain_only: false
seed: 42
task_dataset_name: letters
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.8977163461538461
best_val_epoch: 19
test_acc: 0.896201923076923
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar10-mnist.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: best_val_epoch
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: true
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar10_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar10-mnist.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar10
pretrain_only: false
seed: 42
task_dataset_name: mnist
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.9723333333333334
best_val_epoch: 16
test_acc: 0.9714
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar100-cifar10.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar100_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar100-cifar10.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar100
pretrain_only: false
seed: 42
task_dataset_name: cifar10
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.8867
best_val_epoch: 10
test_acc: 0.8798
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar100-kmnist.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar100_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar100-kmnist.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar100
pretrain_only: false
seed: 42
task_dataset_name: kmnist
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.9846666666666667
best_val_epoch: 15
test_acc: 0.9796
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar100-letters.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar100_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar100-letters.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar100
pretrain_only: false
seed: 42
task_dataset_name: letters
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.9174278846153846
best_val_epoch: 16
test_acc: 0.9126442307692307
37 changes: 37 additions & 0 deletions logs/pretrain_to_task_logs/cifar100-mnist.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
pretrain: best_val_epoch
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: true
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/cifar100_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/cifar100-mnist.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: cifar100
pretrain_only: false
seed: 42
task_dataset_name: mnist
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
task:
best_val_accuracy: 0.98625
best_val_epoch: 15
test_acc: 0.9815
38 changes: 38 additions & 0 deletions logs/pretrain_to_task_logs/imagenet-cifar10.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/imagenet_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/imagenet-cifar10.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: imagenet
pretrain_only: false
seed: 42
task_dataset_name: cifar10
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
use_basic_model: true
task:
best_val_accuracy: 0.898
best_val_epoch: 12
test_acc: 0.8834
38 changes: 38 additions & 0 deletions logs/pretrain_to_task_logs/imagenet-cifar100.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
pretrain: {}
pretrain_to_task_config:
device: cuda
estimate_network_params:
hidden_dim: 256
num_classes: 10
model:
name: resnet18
pretrained: true
need_pretrain: false
paths:
checkpoint: /home/machenike/bmml/DataMetaMap/checkpoints/imagenet_pretrain.pt
checkpoint_task: /home/machenike/bmml/DataMetaMap/checkpoints/best_model.pt
data_dir: /home/machenike/bmml/DataMetaMap/data
save_logs: /home/machenike/bmml/DataMetaMap/logs/pretrain_to_task_logs/imagenet-cifar100.yaml
test_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/test_embeds.parquet
train_parquet: /home/machenike/bmml/DataMetaMap/pretrain_embeddings/train_embeds.parquet
pretrain_dataset_name: imagenet
pretrain_only: false
seed: 42
task_dataset_name: cifar100
training:
batch_size: 128
early_stopping_epochs: 5
epochs: 20
lr: 0.0003
val_split: 0.2
training_task:
batch_size: 128
early_stopping_epochs: 10
epochs: 20
lr: 0.0003
val_split: 0.2
use_basic_model: true
task:
best_val_accuracy: 0.6802
best_val_epoch: 9
test_acc: 0.6789
Loading
Loading