Commit b3ae0a2d authored by kreuzer's avatar kreuzer
Browse files

Aktualisieren scripts/ActorOnly+CEL/main_ActorOnly.py

parent 4f8c1d98

scripts/ActorOnly+CEL/main_ActorOnly.py

+3 −3
Original line number Diff line number Diff line
@@ -65,9 +65,9 @@ m.load_state_dict(best_model_wts) 
# for Critic: save model_actor_only   

torch.save(m.state_dict(), 'model_actor_only_wts.pth')

# set sent_vecs 

train_data.compute_sent_vecs(m)  #

val_data.compute_sent_vecs(m)

test_data.compute_sent_vecs(m)

train_data.compute_sent_vecs(m, 'workspace/students/kreuzer/new_train')  #

val_data.compute_sent_vecs(m, 'workspace/students/kreuzer/new_val')

test_data.compute_sent_vecs(m, 'workspace/students/kreuzer/new_test')



# testing

since = time.time()