bug fixes

ManifoldRG · Apr 17, 2021 · f8c0b0d · f8c0b0d
1 parent 7b16cb0
commit f8c0b0d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/multi_train.py b/multi_train.py
@@ -13,7 +13,11 @@ def multi_train(configs_path="./configs.json"):
  configs = SimpleNamespace(**json.load(open(configs_path)))
  print(configs)
 
- experiment = Experiment(project_name="imagenet-rwc")
+ experiment = Experiment(
+ api_key="ZgD8zJEiZErhwIzPMfZpitMjq",
+ project_name="imagenet-rwc",
+ workspace="ayushm-agrawal",
+ )
  experiment.log_parameters(configs)
 
  average_rmae_dict, average_train_acc, average_test_acc = {

diff --git a/utils/helpers.py b/utils/helpers.py
@@ -62,7 +62,7 @@ def load_model(model_name, training_type, configs):
 
  model = vgg19_bn(in_channels=configs.input_channels,
  num_classes=configs.num_classes)
- elif model_name = "resnet101":
+ elif model_name == "resnet101":
  if training_type == 'pretrained':
  print("Loading pretrained ResNet-101")