@Namespace(value="torch::optim") @Properties(inherit=torch.class) public class SGD extends Optimizer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
SGD(OptimizerParamGroupVector param_groups,
SGDOptions defaults) |
SGD(Pointer p)
Pointer cast constructor.
|
SGD(TensorVector params,
SGDOptions defaults) |
Modifier and Type | Method and Description |
---|---|
void |
load(InputArchive archive)
Deserializes the optimizer state from the given
archive . |
void |
save(OutputArchive archive)
Serializes the optimizer state into the given
archive . |
Tensor |
step() |
Tensor |
step(LossClosure closure)
A loss function closure, which is expected to return the loss value.
|
add_param_group, add_parameters, defaults, param_groups, parameters, size, zero_grad, zero_grad
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public SGD(Pointer p)
Pointer(Pointer)
.public SGD(@ByVal OptimizerParamGroupVector param_groups, @ByVal SGDOptions defaults)
public SGD(@Cast(value={"","std::vector<torch::Tensor>"}) @StdMove TensorVector params, @ByVal SGDOptions defaults)
@ByVal public Tensor step(@ByVal(nullValue="torch::optim::Optimizer::LossClosure(nullptr)") LossClosure closure)
Optimizer
public void save(@ByRef OutputArchive archive)
Optimizer
archive
.public void load(@ByRef InputArchive archive)
Optimizer
archive
.Copyright © 2024. All rights reserved.