Commit 5ff4380a authored by Zhang, Chen's avatar Zhang, Chen
Browse files

fix train param

parent 544cd364

scripts/train_ntrace.py

+4 −4
Original line number Diff line number Diff line
@@ -15,12 +15,12 @@ if __name__ == "__main__": 
    train_params = {

        "cuda_id": 0,

        "n_epochs": 100,  # seems like 50 is the point where training and validation loss diverge

        "n_training": 2_000_000,

        "n_training": 1_000_000,

        "error": 0.07,

        "batch_size": 180,

        "learning_rate": 0.005,

        "weight_decay": 1e-6,

        "optimizer": "Adam",

        "learning_rate": 0.0057929655918116715,

        "weight_decay": 7.198921885462489e-07,

        "optimizer": "SGD",

        "loss": "composite",

        "cache_dir": "data",

        "experiment_name": "nTrace_cu_film_6layer",