summaryrefslogtreecommitdiff
path: root/test/test_nn.py
AgeCommit message (Expand)AuthorFilesLines
2018-09-02improve docker packages, fix bugs, enable tests, enable FFT (#10893)iotamudelta1-48/+239
2018-08-30Fix relying on UB in test_data_parallel_nested_output (#11092)Tongzhou Wang1-2/+2
2018-08-30Add ATen pdist CPU kernel (#10782)Erik Brinkman1-0/+28
2018-08-23Preserve sparse tensor shape and dim invariants, and add scalar tensor suppor...Will Feng1-0/+13
2018-08-22Resubmit #8322 with scipy version checkTongzhou Wang1-4/+411
2018-08-17Fix dropout fused kernel applied in eval mode (#10621)Tongzhou Wang1-0/+5
2018-08-16Add ``buffers()``, ``named_buffers()`` methods. (#10554)Jerry Ma1-36/+62
2018-08-15Revert D9332335: [pytorch][PR] Implements volumetric (5d) affine grid generat...Simon Wang1-389/+1
2018-08-15Move RNN implementations to C++ (#10481)Adam Paszke1-6/+6
2018-08-15Propagate NaN through threshold (#10277)Tongzhou Wang1-0/+2
2018-08-15relax tolerance for two torch.half (float16) tests (#10519)Brian Hart1-2/+2
2018-08-15Implements volumetric (5d) affine grid generation. (#8322)Eli Stevens1-1/+389
2018-08-10Grid sampler: nearest interpolation & reflection padding (#10051)Tongzhou Wang1-119/+244
2018-08-07add fused dropout kernels (#9666)Natalia Gimelshein1-5/+30
2018-08-03use logsigmoid at multilabel_soft_margin_loss, and change output from shape=(...Wei Yang1-3/+5
2018-08-03ctc_loss odds and ends (#10112)Thomas Viehmann1-0/+14
2018-08-01Move grid sampler to ATen (#9961)Tongzhou Wang1-53/+49
2018-08-01Add CELU activation to pytorch (#8551)Xiang Gao1-1/+17
2018-07-31fixed a newly introduced regression in softmax (#10066)Kyle M. Tarplee1-0/+5
2018-07-31fix _pointwise_loss for target gradients (#10018)Roy Li1-0/+6
2018-07-31Add CTC loss (#9628)Thomas Viehmann1-17/+84
2018-07-27Use round-to-negative division when computing output sizes for convolutions i...Owen Anderson1-0/+10
2018-07-18Use _six for inf and nan (#9500)Tongzhou Wang1-4/+3
2018-07-17Implement 2D and 3D alpha_dropout (#9073)tippisum1-12/+23
2018-07-15Add ModuleDict and ParameterDict containers (#8463)Karan Dwivedi1-0/+167
2018-07-11Accumulate MSELoss reduce=True into accreal instead of real (#9287)Richard Zou1-0/+9
2018-07-10Implement grid_sampler in aten (#8929)Roy Li1-0/+7
2018-07-08Update extension docs, fix Fold/Unfold docs (#9239)Tongzhou Wang1-2/+53
2018-07-06Implement nn.functional.interpolate based on upsample. (#8591)Ailing Zhang1-23/+76
2018-07-05Test nn.Module on non-contiguous inputs (#9114)Tongzhou Wang1-4/+19
2018-07-05Turn on UBSAN in the OSS build (#8813)Will Feng1-1/+1
2018-07-02update nn loss tests to use new reduction arg (#9118)Roy Li1-144/+120
2018-07-02detach spectral norm calculated weight in eval mode (#9020)Thomas Viehmann1-0/+6
2018-07-01Fix loading 0.4 BN checkpoints (#9004)Tongzhou Wang1-0/+16
2018-07-01combine size_average and reduce args in loss functions (#8018)Roy Li1-0/+21
2018-06-27Check key should be string in nn.Module.add_module, parameter and buffer (#8960)Karan Dwivedi1-0/+20
2018-06-26Update from Facebook (#8887)Orion Reblitz-Richardson1-1/+2
2018-06-26Add pos_weight argument to nn.BCEWithLogitsLoss (#5660) (#6856)Vadim Velikodniy1-0/+33
2018-06-25throw error when grid_sample is passed unsupported mode (#8884)li-roy1-0/+4
2018-06-24Spectral norm improvements (#8590)Thomas Viehmann1-2/+38
2018-06-22allow output_size to contain None in adaptive pooling methods (#8596)Ailing1-0/+47
2018-06-21Use env var to pass sharding options to test_nn.py (#8727)Will Feng1-26/+18
2018-06-20Shard test_nn to reduce runtime for each test target (#8678)Will Feng1-2/+27
2018-06-18Fix test_rnn_args_check (#8606)Richard Zou1-9/+10
2018-06-14Migrated hardshrink() to ATen and deprecated nn.Hardshrink() (#8117)Wei Yang1-16/+13
2018-06-12[SpectralNorm] don't register original weight as buffer (#8170)Tongzhou Wang1-4/+19
2018-06-07Added backward function for kl_div target (#7839)Wei Yang1-0/+13
2018-06-07Stop BCELoss from returning negative results (#8147)li-roy1-0/+9
2018-06-06skip CUDA memory leak check on Windows altogether (#8213)Tongzhou Wang1-32/+0
2018-06-06Allow parallel_apply to take in list[Tensor] (#8047)Tongzhou Wang1-7/+6
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-27 09:59:33 +0000