Commit 13bea40a authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

changed device to 0

parent 50ee5e00
......@@ -25,7 +25,7 @@ number_of_epochs = 10
loss_log_period = 50
learning_rate_scheduler_step_size = 3
learning_rate_scheduler_gamma = 1e-1
use_last_checkpoint = True
use_last_checkpoint = False
final_model_output_file = "finetuned_alldata.pth.tar"
[NETWORK]
......@@ -46,7 +46,7 @@ number_of_classes = 1
save_model_directory = "saved_models"
model_name = "BrainMapper"
logs_directory = "logs"
device = 1
device = 0
experiments_directory = "experiments"
[EVALUATION]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment