search

tzuhanchang / HyPER

Hypergraph for Particle Event Reconstruction
MIT License
3 stars 0 forks source link

Add additional optimizers #25

Open tzuhanchang opened 4 months ago

tzuhanchang commented 4 months ago

Add torch.nn.AdamW optimizer.

tzuhanchang commented 4 months ago

Add torch.optim.AdamW and torch.optim.SGD. Also, add weight_decay and momentum options, but set their default values to 0.