IUT_Informatique/TP_IA
Archived
1
0
Fork 0
Code Activity

Change data path to a relative path

Browse source
This commit is contained in:
Ethanell 2021-04-03 22:40:37 +02:00
parent 95c2b25268
commit 3e2c776b3e
1 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
main.py
View file

@ -1,5 +1,5 @@
from functools import partial
from os.path import join
from os.path import join, abspath
from numpy.random import randint
from ray import tune
@ -59,13 +59,9 @@ def main(data_root, num_samples=10, max_num_epochs=10, gpus_per_trial=1):
best_checkpoint_dir, "checkpoint"))
best_trained_model.load_state_dict(model_state)
# If Pytorch don't save the end
print("In case saving...")
save(best_trained_model, "/home/flifloo/IA/model.pth")
print("Testing accuracy...")
print(f"Best trial test set accuracy: {test_accuracy(best_trained_model, data_root, device)}")
if __name__ == "__main__":
main("/home/flifloo/IA/data")
main(abspath("data"))